2020年8月3日
pytorch创建自己的图片Dataset
摘要： class mydataset(Dataset): def __init__(self, root, transform=None): super(mydataset,self).__init__() # 所有图片的绝对路径 imgs = os.listdir(root) self.imgs = [ 阅读全文
posted @ 2020-08-03 23:17 林胜联府 阅读(576) 评论(0) 推荐(0) 编辑