Commit c9dddda 1 parent 5444724 commit c9dddda Copy full SHA for c9dddda
File tree 1 file changed +6
-10
lines changed
research/object_detection
1 file changed +6
-10
lines changed Original file line number Diff line number Diff line change 34
34
35
35
def _get_configs_for_model (model_name ):
36
36
"""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' )
47
43
configs = config_util .get_configs_from_pipeline_file (fname )
48
44
return config_util .merge_external_params_with_configs (
49
45
configs ,
You can’t perform that action at this time.
0 commit comments