Skip to content

GitLab

"src/git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "01782c220eea24f74e149c0ab1153e74ba65711b"
Commit c9dddda4 authored by Zhichao Lu's avatar Zhichao Lu Committed by pkulzc
Browse files

Minor fixes to third party code. 

- Fix externally broken test input_test.py
- Add Cython to object detection REQUIRED_PACKAGES

PiperOrigin-RevId: 191338837
parent 5444724e
Hide whitespace changes
Inline Side-by-side
Showing with 6 additions and 10 deletions
research/object_detection/inputs_test.py
View file @ c9dddda4
...@@ -34,16 +34,12 @@ FLAGS = tf.flags.FLAGS ...@@ -34,16 +34,12 @@ 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."""
fname = os.path.join( fname = os.path.join(tf.resource_loader.get_data_files_path(),
FLAGS.test_srcdir, 'samples/configs/' + model_name + '.config')
('google3/third_party/tensorflow_models/' label_map_path = os.path.join(tf.resource_loader.get_data_files_path(),
'object_detection/samples/configs/' + model_name + '.config')) 'data/pet_label_map.pbtxt')
label_map_path = os.path.join(FLAGS.test_srcdir, data_path = os.path.join(tf.resource_loader.get_data_files_path(),
('google3/third_party/tensorflow_models/' 'test_data/pets_examples.record')
'object_detection/data/pet_label_map.pbtxt'))
data_path = os.path.join(FLAGS.test_srcdir,
('google3/third_party/tensorflow_models/'
'object_detection/test_data/pets_examples.record'))
configs = config_util.get_configs_from_pipeline_file(fname) configs = config_util.get_configs_from_pipeline_file(fname)
return config_util.merge_external_params_with_configs( return config_util.merge_external_params_with_configs(
configs, configs,
......
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