登录
首页 » Others » pytorch训练图像分类模型pth转ONNX并测试

pytorch训练图像分类模型pth转ONNX并测试

于 2020-12-03 发布
0 397
下载积分: 1 下载次数: 4

代码说明:

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

下载说明:请别用迅雷下载,失败请重下,重下不扣分!

发表评论

0 个回复

  • 696532资源总数
  • 103709会员总数
  • 38今日下载