【AI实战营第二期】第三次作业提交4班
wyyang23 opened this issue · comments
作业：基于 RTMDet 的气球检测
背景：熟悉目标检测和 MMDetection 常用自定义流程。
任务：
- 基于提供的 notebook，将 cat 数据集换成气球数据集;
- 按照视频中 notebook 步骤，可视化数据集和标签;
- 使用MMDetection算法库，训练 RTMDet 气球目标检测算法，可以适当调参，提交测试集评估指标;
- 用网上下载的任意包括气球的图片进行预测，将预测结果发到群里;
- 按照视频中 notebook 步骤，对 demo 图片进行特征图可视化和 Box AM 可视化，将结果发到群里
- 需提交的测试集评估指标（不能低于baseline指标的50%）
- 目标检测 RTMDet-tiny 模型性能 不低于65 mAP。
数据集
气球数据集可以直接下载https://download.openmmlab.com/mmyolo/data/balloon_dataset.zip
P.S. 同时也欢迎各位选择更复杂的数据集进行训练，如选用同济子豪兄的十类饮料目标检测数据集Drink_284
We recommend using English or English & Chinese for issues so that we could have broader discussion.
