Skip to content

GitLab

"docs/source/en/model_doc/flaubert.md" did not exist on "ec3567fe20e10353ad86310c8750e5100c172124"
Switch branch/tag
Find file BlameHistoryPermalink
  • Harisankar Babu's avatar
    Optionally preprocess segmentation maps for MobileViT (#28420) · d5606378
    Harisankar Babu authored 
    * 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