【代码复现】PyTorch获取读取图像的文件名

背景:论文复现的时候,为了后续操作方便,需要获取到图像的文件名,并按照文件名保存图片

先说结论:pytorch使用dataloader时无法直接获取到图像的文件名,需要实现自定义Dataset类的时候去重写__getitem__方法,在该方法内获取索引并做对应处理后返回

例子:下面是自定义的DataSet类
【代码复现】PyTorch获取读取图像的文件名_第1张图片
【代码复现】PyTorch获取读取图像的文件名_第2张图片

Dataset是用来解决数据从哪里读取以及如何读取的问题。pytorch给定的Dataset是一个抽象类,所有自定义的Dataset都要继承它,并且复写__getitem__()和__len__()类方法,getitem()的作用是接受一个索引,返回一个样本或者标签。

参考链接:pytorch笔记5-数据读取机制DataLoader

你可能感兴趣的:(图像隐写,论文复现,pytorch,人工智能,python)