From 1a33ccc45a2cfe240dde5322436ae1821f23273f Mon Sep 17 00:00:00 2001 From: Elliott Zheng Date: Tue, 2 Jul 2024 20:39:29 +0800 Subject: [PATCH 1/2] fix the RGB order bug in demo.py --- sixdrepnet/demo.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/sixdrepnet/demo.py b/sixdrepnet/demo.py index 16b6a84..7423485 100644 --- a/sixdrepnet/demo.py +++ b/sixdrepnet/demo.py @@ -114,8 +114,7 @@ def parse_args(): y_max = y_max+int(0.2*bbox_width) img = frame[y_min:y_max, x_min:x_max] - img = Image.fromarray(img) - img = img.convert('RGB') + img = Image.fromarray(img[...,::-1]) img = transformations(img) img = torch.Tensor(img[None, :]).to(device) From 34dd4faab8b16b43edbc568f617193cbae0bf620 Mon Sep 17 00:00:00 2001 From: Elliott Zheng Date: Tue, 2 Jul 2024 22:10:28 +0800 Subject: [PATCH 2/2] fix rgb error in face detector --- sixdrepnet/demo.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/sixdrepnet/demo.py b/sixdrepnet/demo.py index 7423485..1d6f448 100644 --- a/sixdrepnet/demo.py +++ b/sixdrepnet/demo.py @@ -94,7 +94,7 @@ def parse_args(): while True: ret, frame = cap.read() - faces = detector(frame) + faces = detector(frame, cv=True) # set cv to True for bgr input, the default value of cv is False for box, landmarks, score in faces: