caffe to pytorch

版权声明:本文为博主原创文章,欢迎转载,转载请注明原文地址、作者信息。 https://blog.csdn.net/hjimce/article/details/82999462

(1)mmdnn docker拉取:docker pull mmdnn/mmdnn:cpu.small

(2)启动docker,并共享主机文件nsfw_model:docker run -it -v /home/hjimce/work/open_nsfw/nsfw_model:/test mmdnn/mmdnn:cpu.small

(3)docker环境模型转换:mmconvert -sf caffe -in deploy.prototxt -iw resnet_50_1by2_nsfw.caffemodel -df pytorch -om caffe_resnet152.pth

猜你喜欢

转载自blog.csdn.net/hjimce/article/details/82999462
今日推荐