-
将Pytorch训练的图像分类模型转换为ONNX格式并进行测试
资源介绍
1. 搭建自己的简单二分类网络,使用pytorch训练和测试;
2. 将pytorch训练的pth模型转换成ONNX,并编码测试;
3. 含训练和测试数据,含训练ok的pth模型和ONNX模型,含完整python和C++实现;
4. 使用方法:首先运行“TrainTestConvertOnnx.py”执行“训练数据读入、模型训练、模型测试、导出onnx”,再运行“TestOnnx.cpp”测试onnx(需要配置OpenCV);