Commit Graph

4 Commits (d8f780c403205c199baf87d05b49181f4790a977)

Author SHA1 Message Date
David de la Iglesia Castro e7240c8cf1 Support custom palette (#157)
* Fix split

* Update tests/test_data/test_dataset.py

Co-authored-by: Jerry Jiarui XU <xvjiarui0826@gmail.com>

Co-authored-by: Jerry Jiarui XU <xvjiarui0826@gmail.com>
2020-09-30 18:02:08 +08:00
David de la Iglesia Castro 51e4cdefc5 Use img_prefix and seg_prefix for loading (#153)
* Use img_prefix and seg_prefix for loading

* flake8

* Fix split
2020-09-25 00:48:16 +08:00
Iago González 7c6fa48411 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
Jiarui XU b2724da80b init commit 2020-07-10 02:39:01 +08:00