登录 注册
当前位置:主页 > 资源下载 > 42 > 将Pytorch训练的图像分类模型转换为ONNX格式并进行测试

将Pytorch训练的图像分类模型转换为ONNX格式并进行测试

  • 更新:2024-07-27 12:43:44
  • 大小:101.11MB
  • 推荐:★★★★★
  • 来源:网友上传分享
  • 类别:深度学习 - 人工智能
  • 格式:RAR

资源介绍

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