Skip to content

Commit c9dddda

Browse files
committed
Minor fixes to third party code.
- Fix externally broken test input_test.py - Add Cython to object detection REQUIRED_PACKAGES PiperOrigin-RevId: 191338837
1 parent 5444724 commit c9dddda

File tree

1 file changed

+6
-10
lines changed

1 file changed

+6
-10
lines changed

research/object_detection/inputs_test.py

+6-10
Original file line numberDiff line numberDiff line change
@@ -34,16 +34,12 @@
3434

3535
def _get_configs_for_model(model_name):
3636
"""Returns configurations for model."""
37-
fname = os.path.join(
38-
FLAGS.test_srcdir,
39-
('google3/third_party/tensorflow_models/'
40-
'object_detection/samples/configs/' + model_name + '.config'))
41-
label_map_path = os.path.join(FLAGS.test_srcdir,
42-
('google3/third_party/tensorflow_models/'
43-
'object_detection/data/pet_label_map.pbtxt'))
44-
data_path = os.path.join(FLAGS.test_srcdir,
45-
('google3/third_party/tensorflow_models/'
46-
'object_detection/test_data/pets_examples.record'))
37+
fname = os.path.join(tf.resource_loader.get_data_files_path(),
38+
'samples/configs/' + model_name + '.config')
39+
label_map_path = os.path.join(tf.resource_loader.get_data_files_path(),
40+
'data/pet_label_map.pbtxt')
41+
data_path = os.path.join(tf.resource_loader.get_data_files_path(),
42+
'test_data/pets_examples.record')
4743
configs = config_util.get_configs_from_pipeline_file(fname)
4844
return config_util.merge_external_params_with_configs(
4945
configs,

0 commit comments

Comments
 (0)