Skip to content

GitLab

Commit d86d4aa6 authored by Fan Yang's avatar Fan Yang Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 413033276
parent b429689b
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
official/vision/beta/serving/semantic_segmentation_test.py
View file @ d86d4aa6
...@@ -31,8 +31,7 @@ from official.vision.beta.serving import semantic_segmentation ...@@ -31,8 +31,7 @@ from official.vision.beta.serving import semantic_segmentation
class SemanticSegmentationExportTest(tf.test.TestCase, parameterized.TestCase): class SemanticSegmentationExportTest(tf.test.TestCase, parameterized.TestCase):
def _get_segmentation_module(self): def _get_segmentation_module(self):
params = exp_factory.get_exp_config('seg_deeplabv3_pascal') params = exp_factory.get_exp_config('mnv2_deeplabv3_pascal')
params.task.model.backbone.dilated_resnet.model_id = 50
segmentation_module = semantic_segmentation.SegmentationModule( segmentation_module = semantic_segmentation.SegmentationModule(
params, batch_size=1, input_image_size=[112, 112]) params, batch_size=1, input_image_size=[112, 112])
return segmentation_module return segmentation_module
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment