Skip to content

GitLab

"tests/git@developer.sourcefind.cn:OpenDAS/apex.git" did not exist on "889d17124c582e588b8b401e69152ef2670091d3"
Commit 08eccced authored by Fan Yang's avatar Fan Yang Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 414061675
parent 997facbb
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
official/vision/beta/serving/export_tflite_lib.py
View file @ 08eccced
...@@ -43,6 +43,8 @@ def create_representative_dataset( ...@@ -43,6 +43,8 @@ def create_representative_dataset(
task = tasks.image_classification.ImageClassificationTask(params.task) task = tasks.image_classification.ImageClassificationTask(params.task)
elif isinstance(params.task, configs.retinanet.RetinaNetTask): elif isinstance(params.task, configs.retinanet.RetinaNetTask):
task = tasks.retinanet.RetinaNetTask(params.task) task = tasks.retinanet.RetinaNetTask(params.task)
elif isinstance(params.task, configs.maskrcnn.MaskRCNNTask):
task = tasks.maskrcnn.MaskRCNNTask(params.task)
elif isinstance(params.task, elif isinstance(params.task,
configs.semantic_segmentation.SemanticSegmentationTask): configs.semantic_segmentation.SemanticSegmentationTask):
task = tasks.semantic_segmentation.SemanticSegmentationTask(params.task) task = tasks.semantic_segmentation.SemanticSegmentationTask(params.task)
......
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