transformers/tests/models/maskformer
NielsRogge 6dc884abc8
[Maskformer] Add MaskFormerSwin backbone (#20344)
* First draft

* Fix backwards compatibility

* More fixes

* More fixes

* Make backbone more general

* Improve backbone

* Improve test

* Fix config checkpoint

* Address comments

* Use model_type

* Address more comments

* Fix special model names

* Remove MaskFormerSwinModel and MaskFormerSwinPreTrainedModel from main init

* Fix typo

* Update backbone

* Apply suggestion

Co-authored-by: Niels Rogge <nielsrogge@Nielss-MacBook-Pro.local>
2022-11-28 20:33:49 +01:00
..
__init__.py
test_feature_extraction_maskformer.py Fix MaskformerFeatureExtractor (#20100) 2022-11-15 16:00:37 +01:00
test_modeling_maskformer_swin.py [Maskformer] Add MaskFormerSwin backbone (#20344) 2022-11-28 20:33:49 +01:00
test_modeling_maskformer.py Fix test_inference_instance_segmentation_head (#17872) 2022-06-24 19:36:45 +02:00