CV图像dataset可以这样做
收藏
图像dataset通常验证结束后，希望显示图片等信息。但是getitem返回的image数据多是经过归一化、转置等等处理过的数据，那么可以设置一方法，通过返回数据集image路径或者原始image数据来显示了。
例如：
def get_origin_data(self):
return self.data
1
收藏
请登录后评论
厉害！
也可以在__getitem__里设置一个开关，设置是否使用数据增强等操作
这样还能在训练集和验证集上重用