brocolli (已弃用,此仓库不再维护)
基于 torch fx 的 PyTorch 模型转换器,包括 pytorch2caffe 和 pytorch2onnx。 基于 torch fx 的 PyTorch 模型量化器。
安装
pip install brocolli
使用方法
-
torch2caffe
- caffe 安装
pip install brocolli-caffe
import torchvision.models as models from brocolli.converter.pytorch_caffe_parser import PytorchCaffeParser net = models.alexnet(pretrained=False) x = torch.rand(1, 3, 224, 224) pytorch_parser = PytorchCaffeParser(net, x) pytorch_parser.convert() pytorch_parser.save('alexnet')
运行此脚本直到屏幕上显示 "accuracy test passed",然后您就可以在当前文件夹下获得 alexnet.caffemodel 和 alexnet.prototxt 文件。
-
torch2onnx
import torchvision.models as models from brocolli.converter.pytorch_onnx_parser import PytorchOnnxParser net = models.alexnet(pretrained=False) x = torch.rand(1, 3, 224, 224) pytorch_parser = PytorchOnnxParser(net, x) pytorch_parser.convert() pytorch_parser.save('alexnet.onnx')
运行此脚本直到屏幕上显示 "accuracy test passed",然后您就可以在当前文件夹下获得 alexnet.onnx 文件。
联系方式
QQ群:597059928
表示支持
如果这个项目对您有帮助,请给个 🌟 吧~