Support to train subsets of coco dataset (#2340)
* support to train subsets of coco dataset * add template config * Add docstrings * Refactor (classes): modity API to be consistent with PR 2408 * Add support for buildin datasets * Merge V2.0 and pass all unittest * Fix (template configs) * Refactor (subset): change function names * Refactor (custom_classes): add subset check * Filter instance during training by self.CLASSES * Fix comments * Raise error when classes is in unsupport type * Refactor: put call of get_subset_by_classes in custom.py * Modify test_dataset to pass CI * Fix custom_classes bug * rename clean_data_infos to subset_data_infos
Showing
- configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco-person-bicycle-car.py 8 additions, 0 deletions...er_rcnn/faster_rcnn_r50_fpn_1x_coco-person-bicycle-car.py
- configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco-person.py 6 additions, 0 deletionsconfigs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco-person.py
- mmdet/datasets/cityscapes.py 2 additions, 0 deletionsmmdet/datasets/cityscapes.py
- mmdet/datasets/coco.py 38 additions, 10 deletionsmmdet/datasets/coco.py
- mmdet/datasets/custom.py 39 additions, 5 deletionsmmdet/datasets/custom.py
- mmdet/datasets/xml_style.py 21 additions, 0 deletionsmmdet/datasets/xml_style.py
- mmdet/models/backbones/resnet.py 5 additions, 1 deletionmmdet/models/backbones/resnet.py
- tests/test_dataset.py 69 additions, 0 deletionstests/test_dataset.py
- tests/test_datasets.py 0 additions, 26 deletionstests/test_datasets.py
- tests/test_wrappers.py 10 additions, 5 deletionstests/test_wrappers.py
Loading
Please register or sign in to comment