Codeformer报错，请大神帮忙看一下

iuranus · 发表于 2024-3-24 11:51:01

星级打分

1
2
3
4
5

平均分:NAN 参与人数:0 我的评分:未评

开始处理...
Face detection model: retinaface_resnet50
[1/1] 正在处理: cover.jpg
Traceback (most recent call last):
  File "CodeFormer/inference_codeformer.py", line 209, in <module>
only_center_face=args.only_center_face, resize=640, eye_dist_threshold=5)
  File "E:\codeFormer-GUI_20240114\_internal\CodeFormer\facelib\utils\face_restoration_helper.py", line 154, in get_face_landmarks_5
bboxes = self.face_det.detect_faces(input_img)
  File "E:\codeFormer-GUI_20240114\_internal\CodeFormer\facelib\detection\retinaface\retinaface.py", line 209, in detect_faces
loc, conf, landmarks, priors = self.__detect_faces(image)
  File "E:\codeFormer-GUI_20240114\_internal\CodeFormer\facelib\detection\retinaface\retinaface.py", line 156, in __detect_faces
loc, conf, landmarks = self(inputs)
  File "E:\codeFormer-GUI_20240114\_internal\Python37\lib\site-packages\torch\nn\modules\module.py", line 1194, in _call_impl
return forward_call(*input, **kwargs)
  File "E:\codeFormer-GUI_20240114\_internal\CodeFormer\facelib\detection\retinaface\retinaface.py", line 121, in forward
out = self.body(inputs)
  File "E:\codeFormer-GUI_20240114\_internal\Python37\lib\site-packages\torch\nn\modules\module.py", line 1194, in _call_impl
return forward_call(*input, **kwargs)
  File "E:\codeFormer-GUI_20240114\_internal\Python37\lib\site-packages\torchvision\models\_utils.py", line 69, in forward
x = module(x)
  File "E:\codeFormer-GUI_20240114\_internal\Python37\lib\site-packages\torch\nn\modules\module.py", line 1194, in _call_impl
return forward_call(*input, **kwargs)
  File "E:\codeFormer-GUI_20240114\_internal\Python37\lib\site-packages\torch\nn\modules\conv.py", line 463, in forward
return self._conv_forward(input, self.weight, self.bias)
  File "E:\codeFormer-GUI_20240114\_internal\Python37\lib\site-packages\torch\nn\modules\conv.py", line 460, in _conv_forward
self.padding, self.dilation, self.groups)
RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED