Moran_2001的博客

原创 pytorch中如何重写dataset-以tif图像数据为例

第二个是__getitem__，这里我认为是最关键的一环，我们需要使用item这个索引来读取我们的数据（一定要用），在前面__init__已经将我们的img_path,label_path送入到了我们的类中，item就是在path中一个迭代的索引，我们无需在自己写for循环来历遍整个数据集，之后再讲图像返回到我们定义的img_transform，label_transform中。使用opencv读取图像我们需要图像的路径，所以定义一个读取图像路径的函数get_img_path。4.验证我们的读取的数据。