VinVL
VinVL copied to clipboard
AssertionError
hello! When I extract image features with VinVL by the command :
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 TEST.OUTPUT_FEATURE True
I get the traceback:
Traceback (most recent call last):
File "tools/test_sg_net.py", line 129, in
I guess the reason is that the file flickr30k/tsv/flickr30k.yaml is not exist, so where can I find this yaml file? Is it necessary?
I also try to delete the line : TEST: ("flickr30k/tsv/flickr30k.yaml",) in the file: vinvl_x152c4.yaml, but after run the code, there is nothing output.
I'm having the same issue. Can someone please explain the cause and solution?