Skip to content

GitLab

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

Internal change 

PiperOrigin-RevId: 414061675
parent ad3a9567
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
official/vision/beta/serving/export_tflite_lib.py
View file @ 65fab8b3
...@@ -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