Can I use official caffe code?

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.34k stars 921 forks source link

Can I use official caffe code? #66

Open bisonliao opened 5 years ago

bisonliao commented 5 years ago

Hi, rich, I compiled your modified caffe in windows successfully, But I got a runtime error: TypeError: No to_python (by-value) converter found for C++ type: class caffe::LayerParameter.

then I try to use my the official caffe installed in my computer, which is from BVLC, it runs ok. so Could you tell me what is the modifications in your caffe version?

thank you.

amoazeni75 commented 3 years ago

Hi, Basically, you have to compile the customized Caffe that is provided on the repo. The reason is that they have some custom layers like "softmax_cross_entropy_loss_layer.cpp" in their resource folder, which needs to be compiled during building Caffe.