YOLOv3-SPP自定义DataSet讲解

文章目录

    • 1.初始化函数
      • 1.1 参数说明
      • 1.2 读取数据
        • 将数据划分到一个个batch中
        • mosaic数据增强调用
        • 创建.shapes文件,保留每张图片的w和h
        • rect启用
        • 缓存标签cache labels
        • 缓存标签images
        • 检测破损images
    • 2. getitem函数
      • 2. 1 训练时 mosaic方法
      • 2. 2 验证时图像处理
      • 2. 3 开启数据增强(augment)
      • 2.3 __getitem__返回值

自定义DataSet数据集:代码在: yolov3_spp-> build_utils->datasets.py ,实现的类为 LoadImageAndLab

你可能感兴趣的:(object,detection,深度学习,计算机视觉,人工智能)