"vscode:/vscode.git/clone" did not exist on "4937d439e947bfcf0983dca7010001aa5777286d"
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,6 +31,9 @@ from object_detection.utils import shape_utils ...@@ -31,6 +31,9 @@ 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():
try:
from official.legacy.image_classification.efficientnet import efficientnet_model
except ModuleNotFoundError:
from official.vision.image_classification.efficientnet import efficientnet_model from official.vision.image_classification.efficientnet import efficientnet_model
_EFFICIENTNET_LEVEL_ENDPOINTS = { _EFFICIENTNET_LEVEL_ENDPOINTS = {
......
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