Skip to content

inisis/brocolli

Repository files navigation

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

image

Show your support

Give a 🌟 if this project helpes~