transformers/tests/models/maskformer
Alara Dirik 7598791c09
Fix MaskFormer failing postprocess tests (#19354)
Ensures post_process_instance_segmentation and post_process_panoptic_segmentation methods return a tensor of shape (target_height, target_width) filled with -1 values if no segment with score > threshold is found.
2022-10-05 23:25:58 +03:00
..
__init__.py Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
test_feature_extraction_maskformer.py Fix MaskFormer failing postprocess tests (#19354) 2022-10-05 23:25:58 +03:00
test_modeling_maskformer.py Fix test_inference_instance_segmentation_head (#17872) 2022-06-24 19:36:45 +02:00