0
点赞
收藏
分享

微信扫一扫

思福迪运维安全管理系统 test_qrcode_b RCE漏洞复现

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

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

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

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

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

举报

相关推荐

0 条评论