tests/models/mobilevit/test_image_processing_mobilevit.py · d5606378851e48053060105f160087694754b610 · chenpangpang / transformers

Optionally preprocess segmentation maps for MobileViT (#28420) · d5606378

Harisankar Babu authored Jan 11, 2024

* 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

d5606378

test_image_processing_mobilevit.py 8.85 KB

Replace test_image_processing_mobilevit.py