a pytorch to caffe model converter, it learns from MMdnn. MMdnn only provides tools for pytorch to onnx conversion, but our tool provides direct conversion from pytorch to caffe
- pip install torch=0.4.0 torchvision
- export PYTHONPATH=$PYTHONPATH:/home/desmond/Github/brocolli/
- python test_pytorch.py
Curently supported layers
- Conv
- PRelu
- MaxPooling
- Sigmoid
- BatchNormalization
- Relu
- Add
- AvgPool
- Flatten
- FullyConnected
- Dropout
- Softmax
- Desmond desmond.yao@buaa.edu.cn