老哥，怎么把ssd从pytorch转caffe。。。

inisis / brocolli

Everything in Torch Fx

MIT License

341 stars 63 forks source link

老哥，怎么把ssd从pytorch转caffe。。。 #8

Closed ZhongyuanW closed 5 years ago

ZhongyuanW commented 5 years ago

如题。。。

inisis commented 5 years ago

按照readme 先从 ssd repo 里面把ssd.py 拷贝到转模型的时候repo 里面。
转模型的时候把 ssd.py 的 156行解注释然后跑。
转成功之后，再跑ssd_layer.py 这时模型就转好了。

ZhongyuanW commented 5 years ago

谢谢！我能转了，但是我ssd的backbone是用的一个mobilenet类，错误信息是找不到mobilenet.bottleneck0.conv1.weights。然后我又仿照vgg的方式把mobilenet结构用list直接写进去，就不会报错了，但是这种就不能读取预训练模型进行训练了，请问有什么解决的办法吗。。。

inisis commented 5 years ago

我的建议，你可以把你的原来模型的model.state_dict.keys()打印出来，这个名字里面应该只有bottleneck0.conv1.weights。你可以想个办法把mobilenet加到key里面去。

ZhongyuanW commented 5 years ago

谢谢大佬！

inisis commented 5 years ago

此 issue 暂时关闭，有任何更新欢迎再次打开。