![](https://img-blog.csdnimg.cn/20201014180756927.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
Pytroch
shunshune
这个作者很懒,什么都没留下…
展开
-
pytorch 加载自己的数据集
pytorch 加载自己的数据集pytorch 加载自己的数据集,需要写一个继承自torch.utils.data中Dataset类,并修改其中的__init__方法、__getitem__方法、__len__方法。默认加载的都是图片,__init__的目的是得到一个包含数据和标签的list,每个元素能找到图片位置和其对应标签。然后用__getitem__方法得到每个元素的图像像素矩阵和标签,返...原创 2019-04-15 17:26:50 · 13522 阅读 · 5 评论 -
windows10 pycharm+Anaconda+pytorch环境配置
1、安装Anaconda我下载的是Anaconda3-5.3.1-Windows-x86_64版本。安装过程中记得将其加入环境变量。2、安装pycharm 社区版。安装的是2018社区版3、anaconda 安装第三方库打开cmd输入conda会显示conda信息表明anaconda安装成功,此时创建虚拟环境pytorch11。虚拟环境的位置会默认选在你的anaconda安装目录下的e...原创 2019-04-16 21:41:54 · 5887 阅读 · 0 评论