scene-graph-TF-release icon indicating copy to clipboard operation
scene-graph-TF-release copied to clipboard

_viz_scene_graph show box defect

Open huang-yuefeng opened this issue 7 years ago • 0 comments

from 100 to 108 in scene-graph-TF-release/lib/datasets/viz.py

for i, bbox in enumerate(rois): if int(labels[i]) == 0 and i not in rel_inds: continue ax.add_patch( plt.Rectangle((bbox[0], bbox[1]), bbox[2] - bbox[0], bbox[3] - bbox[1], fill=False, edgecolor='red', linewidth=3.5) )

actuallly, there are 150 bbox generated by Regression from every rpn roi, each class has one box, so this should be as follows: for i, bbox in enumerate(rois): if int(labels[i]) == 0 and i not in rel_inds: continue$ offset = labels[i]*4 ax.add_patch( plt.Rectangle((bbox[offset+0], bbox[offset+1]), bbox[offset+2] - bbox[offset+0], bbox[offset+3] - bbox[offset+1], fill=False, edgecolor='red', linewidth=3.5) )

this will show right box for highest score.

huang-yuefeng avatar Aug 04 '17 07:08 huang-yuefeng