pytorch学习之旅(一)——自定义数据读取
最近在研究显著性检测,学着使用pytorch框架,以下纯属个人见解,如有错误请指出(一)自定义数据读取首先官方案例:PyTorch读取图片,主要是通过Dataset类,所以先简单了解一下Dataset类。Dataset类作为所有的datasets的基类存在,所有的datasets都需要继承它,类似于C++中的虚基类。class Dataset(object): def __g...