We use the popular MMDetection toolbox for experiments on the MS-COCO dataset with the pre-trained ResNet50 and MobileNetV2 (1.0×) models as the backbones for the detector. We select the mainstream Faster RCNN and Mask R-CNN detectors with Feature Pyramid Networks as the necks to build the basic object detection systems.
Please follow MMDetection on how to prepare the environment and the dataset. Then attach our code to the origin project and modify the config files according to your own path to the pre-trained models and directories to save logs and models.
To train a detector with pre-trained models as backbone:
bash tools/dist_train.sh {path to config file} {ngpus}To evaluate a fine-tuned model:
bash tools/dist_test.sh {path to config file} {path to fine-tuned model} {ngpus} --eval {evaluation metrics} --show| Backbones | Detectors | Params | box AP | mask AP | Config | Google Drive | Baidu Netdisk |
|---|---|---|---|---|---|---|---|
| ResNet50 | Faster R-CNN | 43.80M | 37.4 | - | config | model | model |
| + ODConv (4×) | Faster R-CNN | 108.91M | 39.4 | - | config | model | model |
| MobileNetV2 (1.0×) | Faster R-CNN | 21.13M | 31.8 | - | config | model | model |
| + ODConv (4×) | Faster R-CNN | 29.14M | 35.5 | - | config | model | model |
| ResNet50 | Mask R-CNN | 46.45M | 38.2 | 34.6 | config | model | model |
| + ODConv (4×) | Mask R-CNN | 111.56M | 40.2 | 36.1 | config | model | model |
| MobileNetV2 (1.0×) | Mask R-CNN | 23.78M | 32.4 | 29.8 | config | model | model |
| + ODConv (4×) | Mask R-CNN | 31.80M | 36.0 | 33.0 | config | model | model |