【AI实战营第二期】第三次作业提交06班

[qianlian-mozi]

作业：基于 RTMDet 的气球检测

背景：熟悉目标检测和 MMDetection 常用自定义流程。

任务：

1. 基于提供的 notebook，将 cat 数据集换成气球数据集
2. 按照视频中 notebook 步骤，可视化数据集和标签
3. 使用MMDetection算法库，训练 RTMDet 气球目标检测算法，可以适当调参，提交测试集评估指标
4. 用网上下载的任意包括气球的图片进行预测，将预测结果发到群里
5. 按照视频中 notebook 步骤，对 demo 图片进行特征图可视化和 Box AM 可视化，将结果发到群里 需提交的测试集评估指标（不能低于baseline指标的50%）

• 目标检测 RTMDet-tiny 模型结果的 mAP 不低于 65 数据集 气球数据集可以直接下载 https://download.openmmlab.com/mmyolo/data/balloon_dataset.zip

同时也欢迎各位选择更复杂的数据集进行训练，可以选用一下 10 类的饮料数据集。 https://github.com/TommyZihao/Train_Custom_Dataset/tree/main/%E7%9B%AE%E6%A0%87%E6%A3%80%E6%B5%8B/%E7%9B%AE%E6%A0%87%E6%A3%80%E6%B5%8B%E6%95%B0%E6%8D%AE%E9%9B%86

[mm-assistant[bot]]

We recommend using English or English & Chinese for issues so that we could have broader discussion.

[qianlian-mozi]

https://github.com/qianlian-mozi/period2_mmdet/tree/main

[ZS520L]

https://github.com/ZS520L/MMHomeworks/blob/main/rtmdet_balloon_tutorial.7z

[yandarick]

https://github.com/yandarick/mmdetect_balloon

[zilong-ding]

https://github.com/zilong-ding/rtmdet-balloon/releases/download/first/rtmdet_balloon.ipynb