search

microsoft / scene_graph_benchmark

image scene graph generation benchmark
MIT License
382 stars 86 forks source link

demo yaml #9

Closed JXT218 closed 3 years ago

JXT218 commented 3 years ago

Can you provide flickr30k/tsv/flickr30k.yaml as specified in the vgattr/vinvl_x152c4.yaml？

python tools/test_sg_net.py --config-file sgg_configs/vgattr/vinvl_x152c4.yaml TEST.IMS_PER_BATCH 2 MODEL.WEIGHT models/vinvl/vinvl_vg_x152c4.pth MODEL.ROI_HEADS.NMS_FILTER 1 MODEL.ROI_HEADS.SCORE_THRESH 0.2 DATA_DIR "../maskrcnn-benchmark-1/datasets1" TEST.IGNORE_BOX_REGRESSION True MODEL.ATTRIBUTE_ON True
pzzhang commented 3 years ago

The DATASETS.TEST field (flickr30k/tsv/flickr30k.yaml) is not used in the visualization process. You can just ignore it.