yangxuntu/SGAE

regarding the COCO Test server scene graphs

Opened this issue · 2 comments

Hello @yangxuntu . Thanks for your work!
I'm using the data you provided for the COCO Online Testing server: coco_pred_sg_test.npy. May I know which dictionary is associated for this? I need the scene graphs of the images, just like rela_matrix in coco_img_sg. It seems that the rela_matrix is the same as rela_pred in the file coco_pred_sg_test.npy.

However, when I use the coco_pred_sg_rela.npy dictionary, the results are wierd, as seen below:

sgs= np.load('coco_pred_sg_test.npy')[()]
for row in sgs['275']['rela_pred']:
    print("{0}-{1}-{2}".format(sg_dict[row[0]], sg_dict[row[2]], sg_dict[row[1]]))
shade-clock-monitor
back-clock-shade
clock-curtain-back
clock-back-curtain
clock-curtain-back
back-man-shade
clock-build-back
build-back-keyboard
clock-build-door
shade-back-monitor
headlight-bike-back
build-keyboard-headlight

Any help is appreciated.

@yangxuntu I meet the same question. Have u checked it now?