PytorchToCaffe工具demo测试

pytorch模型转caffe模型的项目地址:

https://github.com/xxradon/PytorchToCaffe

环境

anaconda3(python3.5),编译好的pycaffe,cuda10.1,cudnn 7.6 ,pytorch 1.5.1,torchvision 0.6.1

测试:

vscode中打开项目:

在example文件夹下就是几种不同的网络的转换测试demo,如:

首先测试一下vgg_19_pytorch_to_caffe.py,直接运行:

会报错:

self.bottom.extend(bottom)
TypeError: None has type NoneType, but expected one of: bytes, unicode

网络上下载了一个其他的pytorch-vgg实现,用这个demo测试一下,可以转换成,经过排查,这个demo里的vgg在forward的时候使用的是:avgpool+flatten实现

而对比另外一个实现:

使用的是view代替flatten,转换成功:

使用netron打开查看网络结构:

所以这个错误应该都是由于找不到该层的实现导致的,后来在github的问答中区也找到了相同的答案:

猜你喜欢

转载自blog.csdn.net/sinat_33486980/article/details/118030120