SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Fix image segmentation pipeline errors, resolve backward compatibility issues (#19768)

Browse Source 
* Fix panoptic segmentation and pipeline
* Update ImageSegmentationPipeline tests and reenable test_small_model_pt
* Resolve backward compatibility issues
This commit is contained in:
Alara Dirik
2022-10-21 18:09:58 +03:00
committed by GitHub
parent b58d4f70f6
commit cca51aa151
5 changed files with 113 additions and 137 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/models/maskformer/test_feature_extraction_maskformer.py
View File

@@ -399,13 +399,11 @@ class MaskFormerFeatureExtractionTest(FeatureExtractionSavingTestMixin, unittest
self.assertEqual(segmentation[0].shape, target_sizes[0])
@unittest.skip("Fix me Alara!")
def test_post_process_panoptic_segmentation(self):
feature_extractor = self.feature_extraction_class(num_labels=self.feature_extract_tester.num_classes)
outputs = self.feature_extract_tester.get_fake_maskformer_outputs()
segmentation = feature_extractor.post_process_panoptic_segmentation(outputs, threshold=0)
print(len(segmentation))
print(self.feature_extract_tester.batch_size)
self.assertTrue(len(segmentation) == self.feature_extract_tester.batch_size)
for el in segmentation:
self.assertTrue("segmentation" in el)