qq_39596086的博客

原创 python中pytorch获取数据集

需要自己创建一个接受数据集的类，然后该类需要对Dataset进行继承的操作，进行继承操作之后需要重写__init__方法和getitem方法。os.listdir方法可以将self.path路径下的图片名称转换成一个列表。这个包需要从torch.utils.data包中进行导入。要是进行数据集的拼接操作的话需要对len方法进行重写。pytorch中获取数据集需要用到Dataset。然后就是创建实例，获取数据集。这些操作都可以在控制台进行。