transformers/tests/models/maskformer
NielsRogge b610c47f89
[MaskFormer] Add support for ResNet backbone (#20483)
* Add SwinBackbone

* Add hidden_states_before_downsampling support

* Fix Swin tests

* Improve conversion script

* Add id2label mappings

* Add vistas mapping

* Update comments

* Fix backbone

* Improve tests

* Extend conversion script

* Add Swin conversion script

* Fix style

* Revert config attribute

* Remove SwinBackbone from main init

* Remove unused attribute

* Use encoder for ResNet backbone

* Improve conversion script and add integration test

* Apply suggestion

Co-authored-by: Niels Rogge <nielsrogge@Nielss-MacBook-Pro.local>
2022-12-07 09:42:38 +01:00
..
__init__.py Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
test_feature_extraction_maskformer.py Add segmentation + object detection image processors (#20160) 2022-11-30 10:24:03 +00:00
test_modeling_maskformer_swin.py [Maskformer] Add MaskFormerSwin backbone (#20344) 2022-11-28 20:33:49 +01:00
test_modeling_maskformer.py [MaskFormer] Add support for ResNet backbone (#20483) 2022-12-07 09:42:38 +01:00