transformers/tests/maskformer
Francesco Saverio Zuppichini c4deb7b3ae
Feature Extractor accepts segmentation_maps (#15964)
* feature extractor accepts

* resolved conversations

* added examples in test for ADE20K

* num_classes -> num_labels

* Apply suggestions from code review

Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>

* resolving conversations

* resolving conversations

* removed ADE

* CI

* minor changes in conversion script

* reduce_labels in feature extractor

* minor changes

* correct preprocess for instace segmentation maps

* minor changes

* minor changes

* CI

* debugging

* better padding

* going to update labels inside the model

* going to update labels inside the model

* minor changes

* tests

* removed changes in feature_extractor_utils

* conversation

* conversation

* example in feature extractor

* more docstring in modeling

* test

* make style

* doc

Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>
2022-03-30 18:46:51 +02:00
..
__init__.py Maskformer (#15682) 2022-03-02 15:48:20 +01:00
test_feature_extraction_maskformer.py Feature Extractor accepts segmentation_maps (#15964) 2022-03-30 18:46:51 +02:00
test_modeling_maskformer.py Feature Extractor accepts segmentation_maps (#15964) 2022-03-30 18:46:51 +02:00