"git@developer.sourcefind.cn:OpenDAS/ollama.git" did not exist on "8cbd3e7510d8e30830e695da2de41c687c02e4ce"
Commit 84c0e81f authored by Fan Yang's avatar Fan Yang Committed by TF Object Detection Team
Browse files

Update dependency on official/vision/image_classification.

PiperOrigin-RevId: 420895912
parent 10806935
...@@ -3,7 +3,7 @@ ...@@ -3,7 +3,7 @@
[TOC] [TOC]
To use your own dataset in TensorFlow Object Detection API, you must convert it To use your own dataset in TensorFlow Object Detection API, you must convert it
into the [TFRecord file format](https://www.tensorflow.org/tutorials/load_data/tfrecord). into the [TFRecord file format](https://www.tensorflow.org/api_guides/python/python_io#tfrecords_format_details).
This document outlines how to write a script to generate the TFRecord file. This document outlines how to write a script to generate the TFRecord file.
## Label Maps ## Label Maps
......
...@@ -31,7 +31,10 @@ from object_detection.utils import shape_utils ...@@ -31,7 +31,10 @@ from object_detection.utils import shape_utils
from object_detection.utils import tf_version from object_detection.utils import tf_version
# pylint: disable=g-import-not-at-top # pylint: disable=g-import-not-at-top
if tf_version.is_tf2(): if tf_version.is_tf2():
from official.vision.image_classification.efficientnet import efficientnet_model try:
from official.legacy.image_classification.efficientnet import efficientnet_model
except ModuleNotFoundError:
from official.vision.image_classification.efficientnet import efficientnet_model
_EFFICIENTNET_LEVEL_ENDPOINTS = { _EFFICIENTNET_LEVEL_ENDPOINTS = {
1: 'stack_0/block_0/project_bn', 1: 'stack_0/block_0/project_bn',
......
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