夏侯南溪搭建目标检测模型——数据读取篇

Posted 2021-09-13 songyuc

1 介绍

在PyTorch中定义自己的数据集类，需要继承父类Dataset，

自定义的Dataset类需要实现以下三个函数：

init()：用来实现初始化的操作；

len()：用来返回数据集的样本数目；

getitem()： to support the indexing such that dataset[i] can be used to get i-th sample；

1.0 运行机制

需要注意的是，Dataset类跟Dataloader之间的耦合关系，是否直接调用的关系还需要进行测试；
如果测试符合预期的话，需要对初始化Dataset类的代码进行修改；