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

2023-12-26 17:12:13

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

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

例子：下面是自定义的DataSet类
在这里插入图片描述
在这里插入图片描述

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

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

文章来源:https://blog.csdn.net/qq_40859587/article/details/135223229
本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。