File tree Expand file tree Collapse file tree 2 files changed +5
-2
lines changed
research/object_detection Expand file tree Collapse file tree 2 files changed +5
-2
lines changed Original file line number Diff line number Diff line change 3
3
[ TOC]
4
4
5
5
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 ) .
7
7
This document outlines how to write a script to generate the TFRecord file.
8
8
9
9
## Label Maps
Original file line number Diff line number Diff line change 31
31
from object_detection .utils import tf_version
32
32
# pylint: disable=g-import-not-at-top
33
33
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
35
38
36
39
_EFFICIENTNET_LEVEL_ENDPOINTS = {
37
40
1 : 'stack_0/block_0/project_bn' ,
You can’t perform that action at this time.
0 commit comments