"git@developer.sourcefind.cn:OpenDAS/mmdetection3d.git" did not exist on "ba492be7ea3aa5dbae420a190377496127e767b9"
Commit fe6536b7 authored by Cameron Rudnick's avatar Cameron Rudnick Committed by pkulzc
Browse files

Updated model_lib to use min_score_threshold and max_num_boxes_to_visualize from the eval config.

Updated model_lib to use min_score_threshold and max_num_boxes_to_visualize from the eval config.
parent 8adb303f
...@@ -378,8 +378,8 @@ def create_model_fn(detection_model_fn, configs, hparams, use_tpu=False): ...@@ -378,8 +378,8 @@ def create_model_fn(detection_model_fn, configs, hparams, use_tpu=False):
if not use_tpu and use_original_images: if not use_tpu and use_original_images:
detection_and_groundtruth = ( detection_and_groundtruth = (
vis_utils.draw_side_by_side_evaluation_image( vis_utils.draw_side_by_side_evaluation_image(
eval_dict, category_index, max_boxes_to_draw=20, eval_dict, category_index, max_boxes_to_draw=eval_config.max_num_boxes_to_visualize,
min_score_thresh=0.2, min_score_thresh=eval_config.min_score_threshold,
use_normalized_coordinates=False)) use_normalized_coordinates=False))
img_summary = tf.summary.image('Detections_Left_Groundtruth_Right', img_summary = tf.summary.image('Detections_Left_Groundtruth_Right',
detection_and_groundtruth) detection_and_groundtruth)
......
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