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

Replace reduce_labels with do_reduce_labels (#21218)

Browse Source 
* Replace reduce_labels with do_reduce_labels

* Replace only for __init__ and preprocess

* Update tests
This commit is contained in:
amyeroberts
2023-01-23 17:21:33 +00:00
committed by GitHub
parent 1eda4a4102
commit 354ea44340
5 changed files with 75 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/models/maskformer/test_image_processing_maskformer.py
View File

@@ -53,7 +53,7 @@ class MaskFormerFeatureExtractionTester(unittest.TestCase):
image_mean=[0.5, 0.5, 0.5],
image_std=[0.5, 0.5, 0.5],
num_labels=10,
reduce_labels=True,
do_reduce_labels=True,
ignore_index=255,
):
self.parent = parent
@@ -74,7 +74,7 @@ class MaskFormerFeatureExtractionTester(unittest.TestCase):
self.height = 3
self.width = 4
self.num_labels = num_labels
self.reduce_labels = reduce_labels
self.do_reduce_labels = do_reduce_labels
self.ignore_index = ignore_index
def prepare_feat_extract_dict(self):
@@ -86,7 +86,7 @@ class MaskFormerFeatureExtractionTester(unittest.TestCase):
"image_std": self.image_std,
"size_divisor": self.size_divisor,
"num_labels": self.num_labels,
"reduce_labels": self.reduce_labels,
"do_reduce_labels": self.do_reduce_labels,
"ignore_index": self.ignore_index,
}

8
tests/models/oneformer/test_image_processing_oneformer.py
View File

@@ -69,7 +69,7 @@ class OneFormerImageProcessorTester(unittest.TestCase):
image_mean=[0.5, 0.5, 0.5],
image_std=[0.5, 0.5, 0.5],
num_labels=10,
reduce_labels=False,
do_reduce_labels=False,
ignore_index=255,
repo_path="shi-labs/oneformer_demo",
class_info_file="ade20k_panoptic.json",
@@ -97,7 +97,7 @@ class OneFormerImageProcessorTester(unittest.TestCase):
self.height = 3
self.width = 4
self.num_labels = num_labels
self.reduce_labels = reduce_labels
self.do_reduce_labels = do_reduce_labels
self.ignore_index = ignore_index
def prepare_feat_extract_dict(self):
@@ -108,7 +108,7 @@ class OneFormerImageProcessorTester(unittest.TestCase):
"image_mean": self.image_mean,
"image_std": self.image_std,
"num_labels": self.num_labels,
"reduce_labels": self.reduce_labels,
"do_reduce_labels": self.do_reduce_labels,
"ignore_index": self.ignore_index,
"class_info_file": self.class_info_file,
"metadata": self.metadata,
@@ -180,7 +180,7 @@ class OneFormerImageProcessingTest(FeatureExtractionSavingTestMixin, unittest.Te
self.assertTrue(hasattr(image_processor, "num_text"))
self.assertTrue(hasattr(image_processor, "repo_path"))
self.assertTrue(hasattr(image_processor, "metadata"))
self.assertTrue(hasattr(image_processor, "reduce_labels"))
self.assertTrue(hasattr(image_processor, "do_reduce_labels"))
def test_batch_feature(self):
pass