分类:DataSet
2021
07-22
2021
07-09
目录一、前言二、Dataset三、DatasetLoader四、将Dataset数据和标签放在GPU上(代码执行顺序出错则会有bug)一、前言确保安装scikit-imagenumpy二、Dataset一个例子:#导入需要的包importtorchimporttorch.utils.data.datasetasDatasetimportnumpyasnp#编造数据Data=np.asarray([[1,2],[3,4],[5,6],[7,8]])Label=np.asarray([[0],[1],[0],[2]])#数据[1,2],对应的标签是[0],数据[3,4],对应的标签...
继续阅读 >
2021
03-05