Now only for wheat detection task(single class detection), I have no idea to load multi-class labels after mosaic. May fix it later.
Support SOTA model DetectoRS
Please check configs/detectors/detectors_cascade_rcnn_r50_1x_coco.py/train_pipeline
Please check mmdet/datasets/mosaic.py mosaiccoco.py my_mosaic.py
Please check mmdet/datasets/pipelines/loading
Colab Training Notebook
Replace LoadImageFromFile and LoadAnnotations with LoadMosaicImageAndAnnotations in train_pipeline
https://github.com/open-mmlab/mmdetection