Awesome

brocolli (Deprecated, this repo is no longer maintained)

torch fx based pytorch model converter, including pytorch2caffe, pytorch2onnx.
torch fx based pytorch model quantizier.

Installation

pip install brocolli

How to use

• torch2caffe

  • caffe installation

  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')
  

  run this script until you see "accuracy test passed" on screen, then you can get alexnet.caffemodel and alexnet.prototxt under under current folder.

• 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')
  

  run this script until you see "accuracy test passed" on screen, then you can get alexnet.onnx under current folder.

Contact

QQ Group: 597059928

Show your support

Give a 🌟 if this project helpes~