Skip to content

Commit 84c0e81

Browse files
fyangfTF Object Detection Team
authored andcommitted
Update dependency on official/vision/image_classification.
PiperOrigin-RevId: 420895912
1 parent 1080693 commit 84c0e81

File tree

2 files changed

+5
-2
lines changed

2 files changed

+5
-2
lines changed

research/object_detection/g3doc/using_your_own_dataset.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
[TOC]
44

55
To use your own dataset in TensorFlow Object Detection API, you must convert it
6-
into the [TFRecord file format](https://www.tensorflow.org/tutorials/load_data/tfrecord).
6+
into the [TFRecord file format](https://www.tensorflow.org/api_guides/python/python_io#tfrecords_format_details).
77
This document outlines how to write a script to generate the TFRecord file.
88

99
## Label Maps

research/object_detection/models/ssd_efficientnet_bifpn_feature_extractor.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,10 @@
3131
from object_detection.utils import tf_version
3232
# pylint: disable=g-import-not-at-top
3333
if tf_version.is_tf2():
34-
from official.vision.image_classification.efficientnet import efficientnet_model
34+
try:
35+
from official.legacy.image_classification.efficientnet import efficientnet_model
36+
except ModuleNotFoundError:
37+
from official.vision.image_classification.efficientnet import efficientnet_model
3538

3639
_EFFICIENTNET_LEVEL_ENDPOINTS = {
3740
1: 'stack_0/block_0/project_bn',

0 commit comments

Comments
 (0)