1adrianb / 1adrianb/face-alignment

Detector fails with gray images

Đang mở
#154 2 bình luận 1 reaction 0 người được giao Xem trên GitHub
Ngôn ngữ chính
Python
Star
7.5k
Fork
1.4k
Chỉ số merge pull request
Không có pull request nào được merge trong 30 ngày

Mô tả

When dealing with single channel images, the following error was raised:
`Traceback (most recent call last):
File "face_align.py", line 5, in
lms = fa.get_landmarks_from_directory(dir)
File "/usr/local/lib/python3.5/dist-packages/face_alignment/api.py", line 223, in get_landmarks_from_directory
detected_faces = self.face_detector.detect_from_directory(path, extensions, recursive, show_progress_bar)
File "/usr/local/lib/python3.5/dist-packages/face_alignment/detection/core.py", line 97, in detect_from_directory
predictions[image_path] = self.detect_from_image(image_path)
File "/usr/local/lib/python3.5/dist-packages/face_alignment/detection/sfd/sfd_detector.py", line 51, in detect_from_image
bboxlist = detect(self.face_detector, image, device=self.device)
File "/usr/local/lib/python3.5/dist-packages/face_alignment/detection/sfd/detect.py", line 20, in detect
img = img - np.array([104, 117, 123])
ValueError: operands could not be broadcast together with shapes (398,377) (3,)`

Hướng dẫn đóng góp

Chưa lập chỉ mục được hướng dẫn đóng góp cho kho mã nguồn này

Hướng nghiên cứu

The error occurs in face_alignment/detection/sfd/detect.py line 20 where a single-channel image cannot subtract a 3-channel array. Start by reading the detect function and the SFD detector's image loading logic. Check how images are converted to RGB and handle grayscale inputs. Run the failing test with a grayscale image to reproduce the error, then modify the preprocessing step to support single-channel images.

Do mô hình lập chỉ mục viết ra từ nội dung của issue.

Đánh giá

Công nghệ
python, pytorch
Lĩnh vực
computer-vision
Loại issue
Lỗi
Độ khó
3/5
Thời gian dự kiến
1-2 ngày
Mức độ hoạt động
Đình trệ
Độ rõ ràng
Đặc tả rõ ràng
Mức phù hợp với người mới
45/100

Nhận issue mới trong hộp thư của bạn

Bản tóm tắt ngắn những issue GitHub phù hợp với người mới.