Commit 07ae5c96 authored by Zhichao Lu's avatar Zhichao Lu Committed by lzc5123016
Browse files

Minor fixes to object detection.

PiperOrigin-RevId: 187744580
parent 1491094f
...@@ -53,7 +53,7 @@ recommended. Read [our paper](https://arxiv.org/abs/1611.10012) for a more ...@@ -53,7 +53,7 @@ recommended. Read [our paper](https://arxiv.org/abs/1611.10012) for a more
detailed discussion on the speed vs accuracy tradeoff. detailed discussion on the speed vs accuracy tradeoff.
To help new users get started, sample model configurations have been provided To help new users get started, sample model configurations have been provided
in the object_detection/samples/model_configs folder. The contents of these in the object_detection/samples/configs folder. The contents of these
configuration files can be pasted into `model` field of the skeleton configuration files can be pasted into `model` field of the skeleton
configuration. Users should note that the `num_classes` field should be changed configuration. Users should note that the `num_classes` field should be changed
to a value suited for the dataset the user is training on. to a value suited for the dataset the user is training on.
......
...@@ -34,7 +34,6 @@ FLAGS = tf.flags.FLAGS ...@@ -34,7 +34,6 @@ FLAGS = tf.flags.FLAGS
def _get_configs_for_model(model_name): def _get_configs_for_model(model_name):
"""Returns configurations for model.""" """Returns configurations for model."""
# TODO(ronnyvotel): Make sure these tests work fine outside google3.
fname = os.path.join( fname = os.path.join(
FLAGS.test_srcdir, FLAGS.test_srcdir,
('google3/third_party/tensorflow_models/' ('google3/third_party/tensorflow_models/'
......
...@@ -13,7 +13,7 @@ ...@@ -13,7 +13,7 @@
# limitations under the License. # limitations under the License.
# ============================================================================== # ==============================================================================
"""Tests for image.understanding.object_detection.core.visualization_utils.""" """Tests for object_detection.utils.visualization_utils."""
import logging import logging
import os import os
......
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