1. 首页
  2. 人工智能
  3. 深度学习
  4. pytorch训练图像分类模型pth转ONNX并测试

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

上传者: 2020-07-22 08:27:23上传 RAR文件 101.11MB 热度 173次
1. 搭建自己的简单二分类网络,使用pytorch训练和测试; 2. 将pytorch训练的pth模型转换成ONNX,并编码测试; 3. 含训练和测试数据,含训练ok的pth模型和ONNX模型,含完整python和C++实现; 4. 使用方法:首先运行“TrainTestConvertOnnx.py”执行“训练数据读入、模型训练、模型测试、导出onnx”,再运行“TestOnnx.cpp”测试onnx(需要配置OpenCV);
下载地址
用户评论
码姐姐匿名网友 2020-07-22 08:27:23

解压不了啊

码姐姐匿名网友 2020-07-22 08:27:23

没解压出class_names.class文件,class_names.class里面具体咋写的??

码姐姐匿名网友 2020-07-22 08:27:23

其实就是用opencv调用onnx模型,请在标题或简介中写清楚,这并不是我需要的,不过是个很好的例子。但积分太贵,不值。

码姐姐匿名网友 2020-07-22 08:27:23

压缩坏了骗人的不要下载。。。

码姐姐匿名网友 2020-07-22 08:27:23

博主,你怎么知道cpp对该分类模型输出的解析呢。。。