提交 84c0e81f 编写于 作者: F Fan Yang 提交者: TF Object Detection Team

Update dependency on official/vision/image_classification.

PiperOrigin-RevId: 420895912
上级 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% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册