scene-graph-TF-release
scene-graph-TF-release copied to clipboard
_viz_scene_graph show box defect
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.