Skip to content
Snippets Groups Projects
Unverified Commit fbdb4889 authored by Wenwei Zhang's avatar Wenwei Zhang Committed by GitHub
Browse files

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
parent 1e5e2719
No related branches found
No related tags found
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment