Problem with custom models (*.caffemodel not supported?)

DeadSix27 / waifu2x-converter-cpp

Improved fork of Waifu2X C++ using OpenCL and OpenCV

MIT License

792 stars 86 forks source link

Problem with custom models (*.caffemodel not supported?) #242

Closed Alex20129 closed 4 years ago

Alex20129 commented 4 years ago

i downloaded original models from Nagadomi's repo, but seems like there is no way to use them. when i try to: $ ./waifu2x-converter-cpp --model-dir ../models_waifu2x-caffe/photo -i /home/alex/Pictures/1x -o /home/alex/Pictures/2x i get an error: CUDA: GeForce GTX 1080 Ti Error : couldn't open ../models_waifu2x-caffe/photo//noise1_model.json terminate called after throwing an instance of 'std::runtime_error' what(): model load failed: ../models_waifu2x-caffe/photo//noise1_model.json

YukihoAA commented 4 years ago

waifu2x-conveter-cpp is only accepts torch7 models, especially vgg_7 models. no upconv_7, caffemodel supported.