mmsegmentation/mmseg/datasets
Iago González b8c2f913e4
Add support for custom classes (#71)
* Support for custom classes

* Fix test

* Fix pre-commit

* Add pipeline logic for custom classes

* Fix minor issues, fix test

* Fix issues from PR review

* Fix tests

* Remove palette as str

* Rename old_to_new_ids to label_map

* Test for load_anns

* Remove get_palette function

* fixed temp

* Add subset of palette, remove palette as arg

* minor update

Co-authored-by: Jiarui XU <xvjiarui0826@gmail.com>
2020-09-16 21:33:01 +08:00
..
pipelines Add support for custom classes (#71) 2020-09-16 21:33:01 +08:00
__init__.py init commit 2020-07-10 02:39:01 +08:00
ade.py init commit 2020-07-10 02:39:01 +08:00
builder.py init commit 2020-07-10 02:39:01 +08:00
cityscapes.py Fixed slide inference (#90) 2020-08-25 20:01:01 +08:00
custom.py Add support for custom classes (#71) 2020-09-16 21:33:01 +08:00
dataset_wrappers.py init commit 2020-07-10 02:39:01 +08:00
voc.py init commit 2020-07-10 02:39:01 +08:00