【人脸清晰工具】codeformer新支持40系显卡20240114

noob2021 · 发表于 2022-12-6 12:39:43

我和17楼是同一个问题，有些图片无法开始处理

Face detection model: retinaface_resnet50
Traceback (most recent call last):
File "CodeFormer/inference_codeformer.py", line 182, in <module>
img_shape = InputDflImg.get_shape()
AttributeError: 'NoneType' object has no attribute 'get_shape'

lpk8888 · 发表于 2022-12-6 12:43:16

滚石发表于 2022-12-6 10:32
私聊我QQ563861181，把你测试图片的路径、保存结果路径、软件安装路径发我看下 ...

估计他是直接解压了，文件夹名字里包含了中文，我也是这个问题，改了英文名解决了

2772403575 · 发表于 2022-12-6 13:06:31

报个小bug？我是使用全图清晰化处理3张图片后会报显存不足
RuntimeError: CUDA out of memory. Tried to allocate 1.19 GiB (GPU 0; 4.00 GiB total capacity; 2.25 GiB already allocated; 0 bytes free; 3.15 GiB reserved in total by PyTorch)
虽然显存是比较小，但处理了三张后就爆了我猜测是不是处理完一张图片后没有释放占用（纯猜测）
或许滚石佬可以考虑优化一下小显存使用