
PyTorch 对 YOLO3 的深入解析(第三部分):数据集处理。
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
本章将深入阐述数据处理的相关问题,包括如何读取COCO数据集,以及随后处理自定义数据集。数据预处理的理念,尤其是在YOLOv3中数据集的处理方法,无疑是该章的突出特点。由于YOLOv3对数据集输入有着严格的要求,因此所输入的照片尺寸必须精确为416像素。对于那些无法满足照片尺寸要求的图片,需要进行一系列的调整操作。如果直接采用缩放(resize)操作,则会直接导致信息损失。网络在进行分类学习的过程中,还需要适应照片的尺寸变化,这反过来又会影响训练效果。在YOLOv3中,首先需要调整图像的高度和宽度使其相等,然后进行上采样(upsampling)的缩放操作。同时,还需要修改标签(label)的坐标位置。此外,还需执行随机水平翻转以及随机改变大小的操作。最后,所有图像都将被调整到416像素的大小作为输入。代码class ListDataset(Dataset): #继承Dat
全部评论 (0)
还没有任何评论哟~


