richzhang / colorization

Automatic colorization using deep neural networks. "Colorful Image Colorization." In ECCV, 2016.
http://richzhang.github.io/colorization/
BSD 2-Clause "Simplified" License
3.35k stars 928 forks source link

Demo does not run with caffe-cpu #49

Closed phamconganh closed 6 years ago

phamconganh commented 6 years ago

Hi Richzhang! I run colorization with demo by python3 and caffe cpu-only. When I run 'Test-Time Demo in IPython Notebook':

python3 ./colorize.py -img_in ./demo/imgs/ILSVRC2012_val_00041580.JPEG -img_out ./out.png --gpu 0 WARNING: Logging before InitGoogleLogging() is written to STDERR F0811 16:05:14.388861 419 common.cpp:66] Cannot use GPU in CPU-only Caffe: check mode. Check failure stack trace: Aborted (core dumped)

Can you support me?

phamconganh commented 6 years ago

I found it https://scidle.com/coloring-images-and-video-with-machine-learning/ to solve it