https://github.com/tensorflow/models/pull/5372/files#diff-76df8ca264a059e8a3003851fe4d7849R653
research/object_detection/model_lib.py
def create_train_and_eval_specs(train_input_fn, | |||
name=exporter_name, serving_input_receiver_fn=predict_input_fn) | |||
eval_specs.append( | |||
tf.estimator.EvalSpec( | |||
- | name=eval_spec_name, | ||
+ | name=str(eval_spec_name), | ||
input_fn=eval_input_fn, | |||
steps=None, | |||
exporters=exporter)) | |||
Copy pathView file
2 research/object_detection/models/ssd_mobilenet_v1_fpn_feature_extractor.py
# A modified config of mobilenet v1 that makes it more detection friendly, | |||
def _create_modified_mobilenet_config(): | |||
- | conv_defs = copy.copy(mobilenet_v1.MOBILENETV1_CONV_DEFS) | ||
+ | conv_defs = copy.copy(mobilenet_v1._CONV_DEFS) | ||
conv_defs[-2] = mobilenet_v1.DepthSepConv(kernel=[3, 3], stride=2, depth=512) | |||
conv_defs[-1] = mobilenet_v1.DepthSepConv(kernel=[3, 3], stride=1, depth=256) | |||
return conv_defs |