u010397980的博客

pytorch官网有很好的教程(https://pytorch.org/tutorials/)，相信看完后收获很大，有时候官网国内上不去，我把官网的教程保存下来了(链接: https://pan.baidu.com/s/1e6DOTkjCUVojPuedE2fWiQ 提取码: wjix)。官网给的分类代码(https://github.com/pytorch/examples/blob/mas...

官网的分类代码东西太多，很多功能用不上，于是对官网的代码进行了修改。main()函数对训练模型；predict()函数对训练好的模型进行调用预测。训练图像放在sku_train文件夹中，里面有子文件夹，这些子文件夹的名字分别为各个类，每个子文件夹下为各类的图像。验证图像放在sku_val中，文件结构同上。# -*- coding: utf-8 -*-from __future...

上一篇博客是利用torch自己的datasets工具加载数据。这里加载.txt文件中的数据，txt内容包括图像的路径及标签，中间用空格分开，如下图所示，这个txt文件用python很容易生成。 完整代码如下：较之前的代码多了一个LoadMyDataset类，该类替换类原来的datasets.ImageFolder。此外，还增加了一种RandomErasing随机擦除的数据增强方法(...

pytorch很好用，但是真实部署还是使用c++较多，因此需要用c++调用训练好的pytorch模型。参考官方示例https://github.com/apachecn/pytorch-doc-zh/blob/master/docs/1.0/cpp_export.md安装libtorch。但是官方示例中并没有使用真实的例子，且使用的是cpu版本的。下面用一个真实的例子并用gpu，本示例的所有程...