PaddleDetection数据输入

最新推荐文章于 2024-07-20 22:35:57 发布

烩鳗鱼

最新推荐文章于 2024-07-20 22:35:57 发布

阅读量206

点赞数

分类专栏：笔记文章标签： python 深度学习计算机视觉

本文链接：https://blog.csdn.net/m0_57173620/article/details/120943795

版权

PaddleDetection的数据处理模块在ppdet/data/中，负责加载数据并转化为模型所需的格式。主要包括COCO和VOC数据源，通过load_roidb_and_cname2cid方法解析标注文件，子类需重写实现具体操作。

摘要由CSDN通过智能技术生成

2021SC@SDUSC
PaddleDetection的数据处理模块是一个Python模块，所有代码逻辑在ppdet/data/中，数据处理模块用于加载数据并将其转换成适用于物体检测模型的训练、评估、推理所需要的格式。数据处理模块的主要构成如下架构所示：

 ppdet/data/
  ├── reader.py     # 数据处理模块的总接口
  ├── shared_queue  # 共享内存管理模块
  │   ├── queue.py        # 定义共享内存队列
  │   ├── sharedmemory.py # 负责分配内存
  ├── source  # 数据源管理模块
  │   ├── dataset.py      # 定义数据源基类，各类数据集继承于此
  │   ├── coco.py         # COCO数据集解析与格式化数据
  │   ├── voc.py          # Pascal VOC数据集解析与格式化数据
  │   ├── widerface.py    # WIDER-FACE数据集解析与格式化数据
  ├── tests  # 单元测试模块
  │   ├── test_dataset.py # 对数据集解析、加载等进行单元测试
  │   │   ...
  ├── transform  # 数据预处理模块
  │   ├── batch_operators.py  # 定义各类基于批量数据的预处理算子
  │   ├── op_helper.py    # 预处理算子的辅助函数
  │   ├── operators.py    # 定义各类基于单张图片的预处理算子
  ├── parallel_map.py     # 在多进程/多线程模式中对数据预处理操作进行加速

load_roidb_and_cname2cid()在Dataset基类中并没有做实际的操作，需要在子类中重写实际操作。
roidbs:代表数据源列表。load_roidb_and_cname2cid方法中根据数据集标注文件(anno_path)&#