参考pytorch官网教程(https://pytorch.org/tutorials/beginner/data_loading_tutorial.html),自己针对要用的数据写了数据读取代码。
由于我的图像带有时间顺序,后续要用RNN训练,本来的数据文件存储为"标签/样本标号/按顺序存放的n张图像"。
因此,为了方便写Dataset类,我先把每张图片按顺序把路径和对应的标签存到了csv文件中,再在dataset类里面读取。
下面是这个过程中遇到的一些函数使用问题记录。
路径拼接:
os.path.join(base_path,i):该函数使用时,输入为字符串,无需中间加'/'。
创建多维列表:
img_all=[[]for m in range(2)]:该语句在列表中预留两行空间,通过索引访问,后续用append会从第3行开始加入