From 699154628443b18b846aafc658ea76a738a76236 Mon Sep 17 00:00:00 2001 From: littletomatodonkey Date: Fri, 3 Dec 2021 15:08:18 +0800 Subject: [PATCH] fix mainbody data num and yaml (#4794) --- .../picodet/application/mainbody_detection/README.md | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/configs/picodet/application/mainbody_detection/README.md b/configs/picodet/application/mainbody_detection/README.md index 3bb26fc15..dc75d9f3e 100644 --- a/configs/picodet/application/mainbody_detection/README.md +++ b/configs/picodet/application/mainbody_detection/README.md @@ -14,11 +14,11 @@ PP-ShiTu图像识别任务中，训练主体检测模型时主要用到了以下 | 数据集 | 数据量 | 主体检测任务中使用的数据量 | 场景 | 数据集地址 | | :------------: | :-------------: | :-------: | :-------: | :--------: | -| Objects365 | 170W | 6k | 通用场景 | [地址](https://www.objects365.org/overview.html) | -| COCO2017 | 12W | 5k | 通用场景 | [地址](https://cocodataset.org/) | -| iCartoonFace | 2k | 2k | 动漫人脸检测 | [地址](https://github.com/luxiangju-PersonAI/iCartoonFace) | -| LogoDet-3k | 3k | 2k | Logo检测 | [地址](https://github.com/Wangjing1551/LogoDet-3K-Dataset) | -| RPC | 3k | 3k | 商品检测 | [地址](https://rpc-dataset.github.io/) | +| Objects365 | 1700K | 173k | 通用场景 | [地址](https://www.objects365.org/overview.html) | +| COCO2017 | 118K | 118k | 通用场景 | [地址](https://cocodataset.org/) | +| iCartoonFace | 48k | 48k | 动漫人脸检测 | [地址](https://github.com/luxiangju-PersonAI/iCartoonFace) | +| LogoDet-3k | 155k | 155k | Logo检测 | [地址](https://github.com/Wangjing1551/LogoDet-3K-Dataset) | +| RPC | 54k | 54k | 商品检测 | [地址](https://rpc-dataset.github.io/) | 在实际训练的过程中，将所有数据集混合在一起。由于是主体检测，这里将所有标注出的检测框对应的类别都修改为 `前景` 的类别，最终融合的数据集中只包含 1 个类别，即前景，数据集定义配置可以参考[mainbody_detection.yml](./mainbody_detection.yml)。 @@ -27,4 +27,4 @@ PP-ShiTu图像识别任务中，训练主体检测模型时主要用到了以下 | 模型 | 图像输入尺寸 | mAPval
0.5:0.95 | mAPval
0.5 | 下载地址 | config | | :-------- | :--------: | :---------------------: | :----------------: | :----------------: | :---------------: | -| PicoDet-LCNet_x2_5 | 640*640 | 41.5 | 62.0 | [trained model](https://paddledet.bj.bcebos.com/models/picodet_lcnet_x2_5_640_mainbody.pdparams) | [inference model](https://paddledet.bj.bcebos.com/models/picodet_lcnet_x2_5_640_mainbody_infer.tar) | [log](https://paddledet.bj.bcebos.com/logs/train_picodet_lcnet_x2_5_640_mainbody.log) | [config](./picodet_lcnet_2_5x_640_mainbody.yml) | +| PicoDet-LCNet_x2_5 | 640*640 | 41.5 | 62.0 | [trained model](https://paddledet.bj.bcebos.com/models/picodet_lcnet_x2_5_640_mainbody.pdparams) | [inference model](https://paddledet.bj.bcebos.com/models/picodet_lcnet_x2_5_640_mainbody_infer.tar) | [log](https://paddledet.bj.bcebos.com/logs/train_picodet_lcnet_x2_5_640_mainbody.log) | [config](./picodet_lcnet_x2_5_640_mainbody.yml) | -- GitLab