加载数据集

A custom Dataset class must implement three functions:__init__初始化方法__len__告诉pytorch数据集长度是多少数据集是多大的__getitem__告诉pytorch如何获取数据集当中的某一个数据__init__设置一些变量属性也可以给其他方法使用def __init__(self, annotations_file, img_dir, transformNone, target_transformNone): #这个初始化方法__init__接self,说明这个方法属于这个类类可以看做是工具箱方法可以看做是工具 #标注文件夹图片文件夹图片变换标注变换 self.img_labels pd.read_csv(annotations_file) self.img_dir img_dir #图片文件夹复制给self的图片文件夹这样在其他方法中也可以使用 self.transform transform self.target_transform target_transform__len__def __len__(self): return len(self.img_labels) #返回有多少个标注文件__getitem__def __getitem__(self, idx): #idx指第几个数据 img_path os.path.join(self.img_dir, self.img_labels.iloc[idx, 0]) image decode_image(img_path) label self.img_labels.iloc[idx, 1] if self.transform: image self.transform(image) if self.target_transform: label self.target_transform(label) return image, label #返回数据一般包括图片以及标注PyCharm新建一个项目项目取名字 yy-object-detection-model自定义环境custom environmentselect existingtype:condabase环境课程中开发环境程序中打断点查看image点击view as image,vocyolototensor进行了归一化winshifts电脑截屏