pytorch训练图像分类模型pth转ONNX并测试
于 2020-12-03 发布
0 640
下载积分: 1
下载次数: 4
代码说明：
1. 搭建自己的简单二分类网络，使用pytorch训练和测试； 2. 将pytorch训练的pth模型转换成ONNX，并编码测试； 3. 含训练和测试数据，含训练ok的pth模型和ONNX模型，含完整python和C++实现； 4. 使用方法：首先运行“TrainTestConvertOnnx.py”执行“训练数据读入、模型训练、模型测试、导出onnx”，再运行“TestOnnx.cpp”测试onnx（需要配置OpenCV）；
下载说明：请别用迅雷下载，失败请重下，重下不扣分！
发表评论
