我想使用Interactive-Deep-Colorization，然后尝试从Docker文件构建它。
官方资料库：https://github.com/junyanz/interactive-deep-colorization 码头工人文件：https://hub.docker.com/r/vbisbest/ideepcolor_osx/
但是，我的macOS无法打开该文件。
这样做的时候
docker run -ti --rm -e DISPLAY=192.168.1.13:0 -v /tmp/.X11-unix:/tmp/.X11-unix:ro -v /Users:/root/ideepcolor/osx vbisbest/ideepcolor_osx /bin/sh -c 'cd ideepcolor; python ideepcolor.py --cpu_mode'
我明白了
Traceback (most recent call last):
File "ideepcolor.py", line 57, in <module>
colorModel.prep_net(args.gpu,args.color_prototxt,args.color_caffemodel)
File "/root/ideepcolor/data/colorize_image.py", line 252, in prep_net
if(len(self.net.params[self.pred_ab_layer] [0].data[...].shape)==4 \
KeyError: 'pred_ab'
还有其他/更好的解决方案吗？
我的环境设置如下：