自定义博客皮肤VIP专享

*博客头图:

格式为PNG、JPG,宽度*高度大于1920*100像素,不超过2MB,主视觉建议放在右侧,请参照线上博客头图

请上传大于1920*100像素的图片!

博客底图:

图片格式为PNG、JPG,不超过1MB,可上下左右平铺至整个背景

栏目图:

图片格式为PNG、JPG,图片宽度*高度为300*38像素,不超过0.5MB

主标题颜色:

RGB颜色,例如:#AFAFAF

Hover:

RGB颜色,例如:#AFAFAF

副标题颜色:

RGB颜色,例如:#AFAFAF

自定义博客皮肤

-+
  • 博客(1)
  • 收藏
  • 关注

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

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

2024-04-01 10:05:05 435

空空如也

空空如也

TA创建的收藏夹 TA关注的收藏夹

TA关注的人

提示
确定要删除当前文章?
取消 删除