2424004764 / old_photos_reappear_brilliance Goto Github PK

cd /usr/yifang/photo_restoration
python3.8 run.py --input_folder /usr/yifang/photo_restoration/test_images/old --output_folder /usr/yifang/photo_restoration/output --GPU -1

大佬，带划痕的怎么不行，最后没生成图片。不带划痕的是正常的。
是我这命令不对吗？就把test_images/old换成test_images/old_w_scratch，加了个 --with_scratch。
GPU那用改代码吗，还是文件夹路径没弄对

root@f917cf5388bc:/usr/yifang/photo_restoration# python3.8 run.py --input_folder /usr/yifang/photo_restoration/test_images/old_w_scratch --output_folder /usr/yifang/photo_restoration/output --GPU -1 --with_scratch

Running Stage 1: Overall restoration
initializing the dataloader
/usr/local/lib/python3.8/site-packages/torch/cuda/init.py:52: UserWarning: CUDA initialization: Found no NVIDIA driver on your system. Please check that you have an NVIDIA GPU and installed a driver from http://www.nvidia.com/Download/index.aspx (Triggered internally at /pytorch/c10/cuda/CUDAFunctions.cpp:100.)
return torch._C._cuda_getDeviceCount() > 0
model weights loaded
Traceback (most recent call last):
File "detection.py", line 155, in
main(config)
File "detection.py", line 80, in main
model.to(config.GPU)
File "/usr/local/lib/python3.8/site-packages/torch/nn/modules/module.py", line 600, in to
device, dtype, non_blocking, convert_to_format = torch._C._nn._parse_to(*args, **kwargs)

RuntimeError: Device index must not be negative

You are using NL + Res
Traceback (most recent call last):
File "test.py", line 110, in
input_loader = os.listdir(opt.test_input)

FileNotFoundError: [Errno 2] No such file or directory: '/usr/yifang/photo_restoration/output/stage_1_restore_output/masks/input'

Finish Stage 1 ...

无法运行
docker import face_fix.tar hktalent/face_fix:1.0
docker run --rm -it -v $PWD:/app --name face_fix hktalent/face_fix:1.0 /bin/bash
然后卡死在run上

不知道这个镜像适不适合MacBook Pro？

大佬，有 Dockerfile 提供嘛