tests/pipelines/test_pipelines_image_segmentation.py · 47b9165109dc19ef285179ced9ef856053ebcbc9 · chenpangpang / transformers

Add `ForInstanceSegmentation` models to `image-segmentation` pipelines (#15937) · f4e4ad34

Nicolas Patry authored Mar 09, 2022

* Adding ForInstanceSegmentation to pipelines.

* Last fix `category_id` renamed to `label_id`.

* Can't be none no more.

* No `is_thing_map` anymore.

f4e4ad34

test_pipelines_image_segmentation.py 13.7 KB

Replace test_pipelines_image_segmentation.py