pzzhang/VinVL

How to generate features?

Edwardmark opened this issue · 0 comments

Run the demo with the following:

python -m pdb tools/demo/demo_image.py \
       --config_file sgg_configs/vgattr/vinvl_x152c4.yaml \
       --img_file ./test_ims/641.jpg \
       --save_file output/641.obj.jpg \
       --visualize_attr \
       MODEL.WEIGHT models/vinvl/vinvl_vg_x152c4.pth \
       MODEL.ROI_HEADS.NMS_FILTER 1 \
       MODEL.ROI_HEADS.SCORE_THRESH 0.2 \
       DATA_DIR "./test_ims" \
       TEST.IGNORE_BOX_REGRESSION True \
       MODEL.ATTRIBUTE_ON True \
       TEST.OUTPUT_FEATURE True \
# TEST.IGNORE_BOX_REGRESSION True for RPN, False for detection

Where is the base64 features? I cannot find it.