README.md

pytorchTocaffe

本代码实现pytorch模型转换为caffe模型

1. generate_prototxt.py

生成网络caffe对应的prototxt文件

2. pytorchToCaffe.py

将你的pytorch模型转换为caffe模型

3. test_ponet.py

验证你的pytorch模型，得到输出

4. test_ponet_caffe.py

验证你转换的caffe模型，与pytorch模型结果进行对比