pytorch2caffe

最新推荐文章于 2025-04-08 21:41:45 发布

酸菜包子

最新推荐文章于 2025-04-08 21:41:45 发布

阅读量2.2k

点赞数

本文链接：https://blog.youkuaiyun.com/weixin_38378417/article/details/90699348

版权

博客介绍了基于ssd caffe对PytorchConverter源码进行转换的相关内容。因原作者非基于ssd caffe，需做少许改动，如增加permute层。还给出了caffe源环境不同时的处理方法，包括编译caffe、替换caffe_pb2.py、实现permute层等，同时提及了环境要求。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

大多借鉴于：https://github.com/starimeL/PytorchConverter

但由于原作者不是基于ssd caffe，有少许需要改动，主要是增加permute层

使用作者这个源码进行转换，若caffe源环境不同，需要：

1\编译好自己使用的caffe，如ssd caffe，然后将caffe_pb2.py对源码进行替换。

2\参考ConvertLayer_caffe.py中别的层实现方式，实现permute层：

def Permute(pytorch_layer):
    layer = pb2.LayerParameter()
    layer.type = 'Permute'
    assert len(pytorch_layer.rev_dim_indices) == 4, len(pytorch_layer.rev_dim_indices)
    assert pytorch_layer.rev_dim_indices[0] == 0, pytorch_layer.rev_dim_indices[0]
    layer.permute_param.order.extend([0,2,3,1])
    return layer

并在def build_converter(opts):中增加

3\更改别的numpy()为.cpu().numpy()

PS.环境：

python3 pytorch 0.2.0

caffe需要支持使用层的源

注意torchvision版本

pip3 install torchvision==0.2.0