Use get_data_files_path to access test data.

PiperOrigin-RevId: 190505306

research/object_detection/model_test.py

@@ -32,20 +32,19 @@ from object_detection.builders import model_builder
 from object_detection.core import standard_fields as fields
 from object_detection.utils import config_util
 
-FLAGS = tf.flags.FLAGS
 
 MODEL_NAME_FOR_TEST = model_test_util.SSD_INCEPTION_MODEL_NAME
 
 
 def _get_data_path():
   """Returns an absolute path to TFRecord file."""
-  return os.path.join(FLAGS.test_srcdir, model_test_util.PATH_BASE, 'test_data',
+  return os.path.join(tf.resource_loader.get_data_files_path(), 'test_data',
                       'pets_examples.record')
 
 
 def _get_labelmap_path():
   """Returns an absolute path to label map file."""
-  return os.path.join(FLAGS.test_srcdir, model_test_util.PATH_BASE, 'data',
+  return os.path.join(tf.resource_loader.get_data_files_path(), 'data',
                       'pet_label_map.pbtxt')
 
 

research/object_detection/model_test_util.py

@@ -28,13 +28,12 @@ FLAGS = tf.flags.FLAGS
 
 FASTER_RCNN_MODEL_NAME = 'faster_rcnn_resnet50_pets'
 SSD_INCEPTION_MODEL_NAME = 'ssd_inception_v2_pets'
-PATH_BASE = 'google3/third_party/tensorflow_models/object_detection/'
 
 
 def GetPipelineConfigPath(model_name):
   """Returns path to the local pipeline config file."""
-  return os.path.join(FLAGS.test_srcdir, PATH_BASE, 'samples', 'configs',
-                      model_name + '.config')
+  return os.path.join(tf.resource_loader.get_data_files_path(), 'samples',
+                      'configs', model_name + '.config')
 
 
 def InitializeFlags(model_name_for_test):