1adrianb / 1adrianb/face-alignment

Detector fails with gray images

未关闭
#154 2 条评论 1 个 reaction 已指派 0 人 在 GitHub 查看
主要语言
Python
星标
7.5k
派生
1.4k
PR 合并指标
30 天内没有已合并 PR

描述

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,)`

贡献指南

这个仓库没有索引到贡献指南

调研方向

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.

由索引模型根据 Issue 内容生成。

评估

技术栈
python, pytorch
领域
computer-vision
Issue 类型
缺陷
难度
3/5
预计耗时
1-2 天
活跃度
停滞
描述清晰度
描述清楚
新手友好度
45/100

把新 issue 发到你的邮箱

精选适合新手参与的 GitHub issue 摘要。