rpn score

Author: YunYang1994

4-Object_Detection/RPN/test.py

@@ -34,7 +34,7 @@
     image_data = np.expand_dims(raw_image / 255., 0)
     pred_scores, pred_bboxes = model(image_data)
     pred_scores = tf.nn.softmax(pred_scores, axis=-1)
-    pred_scores, pred_bboxes = decode_output(pred_bboxes, pred_scores, 0.7)
+    pred_scores, pred_bboxes = decode_output(pred_bboxes, pred_scores, 0.9)
     pred_bboxes = nms(pred_bboxes, pred_scores, 0.5)
     plot_boxes_on_image(raw_image, pred_bboxes)
     save_path = os.path.join(prediction_result_path, str(idx)+".jpg")

4-Object_Detection/RPN/utils.py

@@ -101,7 +101,7 @@ def compute_regression(box1, box2):
 
     return target_reg
 
-def decode_output(pred_bboxes, pred_scores, score_thresh=0.):
+def decode_output(pred_bboxes, pred_scores, score_thresh=0.5):
     """
     pred_bboxes shape: [1, 45, 60, 9, 4]
     pred_scores shape: [1, 45, 60, 9, 2]