Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
ModelZoo
ResNet50_tensorflow
Commits
84c0e81f
Commit
84c0e81f
authored
Jan 10, 2022
by
Fan Yang
Committed by
TF Object Detection Team
Jan 10, 2022
Browse files
Update dependency on official/vision/image_classification.
PiperOrigin-RevId: 420895912
parent
10806935
Changes
2
Show whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
5 additions
and
2 deletions
+5
-2
research/object_detection/g3doc/using_your_own_dataset.md
research/object_detection/g3doc/using_your_own_dataset.md
+1
-1
research/object_detection/models/ssd_efficientnet_bifpn_feature_extractor.py
...ection/models/ssd_efficientnet_bifpn_feature_extractor.py
+4
-1
No files found.
research/object_detection/g3doc/using_your_own_dataset.md
View file @
84c0e81f
...
...
@@ -3,7 +3,7 @@
[TOC]
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.
## Label Maps
...
...
research/object_detection/models/ssd_efficientnet_bifpn_feature_extractor.py
View file @
84c0e81f
...
...
@@ -31,6 +31,9 @@ from object_detection.utils import shape_utils
from
object_detection.utils
import
tf_version
# pylint: disable=g-import-not-at-top
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
_EFFICIENTNET_LEVEL_ENDPOINTS
=
{
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment