VCTree-Scene-Graph-Generation icon indicating copy to clipboard operation
VCTree-Scene-Graph-Generation copied to clipboard

generate scene graph for random image without any gt boxes, classes or relations

Open Thoudancer opened this issue 6 years ago • 2 comments

Hello, can we generate scenegraph for random image without any gt boxes, classes or relations using the sgdet model?

if so, what part of code should i change or what file should i prepare?

Thoudancer avatar Sep 12 '19 06:09 Thoudancer

I think you can try adding an if/else in lib/object_detector.py just like the code below which is written by the author of neural-motifs. if self.training and not self.mode == 'gtbox': pred_to_gtbox = bbox_overlaps(box_priors, gt_boxes).data pred_to_gtbox[im_inds.data[:, None] != gt_classes.data[None, :, 0]] = 0.0 max_overlaps, argmax_overlaps = pred_to_gtbox.max(1) rm_obj_labels = gt_classes[:, 1][argmax_overlaps] rm_obj_labels[max_overlaps < 0.5] = 0 else: rm_obj_labels = None

monkey1208 avatar Sep 17 '19 09:09 monkey1208

Hello, can we generate scenegraph for random image without any gt boxes, classes or relations using the sgdet model?

if so, what part of code should i change or what file should i prepare?

hi! have you solved this problem? tks

forbiddenname avatar May 23 '20 06:05 forbiddenname