Skip to content

S
seminar-breakout
Commit 6d5ba6a9 authored by Yuxin Wu's avatar Yuxin Wu
Browse files
Options

fix a missing tensor for visualization (fix #530)

parent 16f32f40
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
examples/FasterRCNN/model.py
View file @ 6d5ba6a9
...@@ -213,6 +213,7 @@ def generate_rpn_proposals(boxes, scores, img_shape): ...@@ -213,6 +213,7 @@ def generate_rpn_proposals(boxes, scores, img_shape):
topk_valid_boxes, topk_valid_boxes,
nms_indices, name='boxes') nms_indices, name='boxes')
final_scores = tf.gather(topk_valid_scores, nms_indices, name='scores') final_scores = tf.gather(topk_valid_scores, nms_indices, name='scores')
tf.sigmoid(final_scores, name='probs') # for visualization
return final_boxes, final_scores return final_boxes, final_scores
......
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