文件名称:PytorchConverter:pytorch模型到caffe和ncnn
文件大小:251KB
文件格式:ZIP
更新时间:2024-05-30 04:31:14
Python
火炬转换器 将Pytorch模型转换为Caffe和 型号范例 来自Torchvision的SqueezeNet 来自Torchvision的DenseNet (with ceiling_mode = True) 移动网 来自作者的AnimeGAN预训练模型( ) 类似于SSD的物体检测网(用于ncnn) UNet(尚无预训练的模型,只有默认初始化) 注意事项 注意Pytorch和Caffe,ncnn之间的合并层ceil_mode的差异 您可以使用所有池化层的ceil_mode = True转换Pytorch模型。 或编译Caffe / ncnn的自定义版本,在池层推理中将floor()替换为ceil()。 Python错误:仅使用Pytorch 0.2.0转换模型 pytorch 0.3.0、0.3.1、0.4.0的更高版本似乎阻止了第三方模型的转换。 请注意,您仍然可
【文件预览】:
PytorchConverter-master
----.gitignore(6B)
----code()
--------ReplaceDenormals.py(312B)
--------run.py(4KB)
--------caffe.proto(59KB)
--------test.py(4KB)
--------ConvertLayer_caffe.py(11KB)
--------ConvertModel.py(13KB)
--------ConvertLayer_ncnn.py(13KB)
--------caffe_pb2.py(253KB)
----ModelFiles()
--------UNet()
--------FaceBoxes()
--------_netG_1()
--------MobileNet()
--------ResNet()
----LICENSE(1KB)
----README.md(2KB)
----TestData()
--------227-3.jpg(21KB)
--------2008_000536.jpg(21KB)
--------2008_001171.jpg(30KB)
--------2008_001601.jpg(34KB)
--------227-2.jpg(28KB)
--------ImageNetLabels.txt(29KB)
--------227.jpg(20KB)
--------2008_001841.jpg(23KB)