transformers/tests/models/mobilevit
Harisankar Babu d560637885
Optionally preprocess segmentation maps for MobileViT (#28420)
* optionally preprocess segmentation maps for mobilevit

* changed pretrained model name to that of segmentation model

* removed voc-deeplabv3 from model archive list

* added preprocess_image and preprocess_mask methods for processing images and segmentation masks respectively

* added tests for segmentation masks based on segformer feature extractor

* use crop_size instead of size

* reverting to initial model
2024-01-11 14:52:14 +00:00
..
__init__.py add MobileViT model (#17354) 2022-06-29 16:07:51 -04:00
test_image_processing_mobilevit.py Optionally preprocess segmentation maps for MobileViT (#28420) 2024-01-11 14:52:14 +00:00
test_modeling_mobilevit.py Update forward signature test for vision models (#27681) 2023-11-27 15:48:17 +01:00
test_modeling_tf_mobilevit.py Make more test models smaller (#25005) 2023-07-24 10:08:47 -04:00