search

pzzhang / VinVL

project page for VinVL
349 stars 25 forks source link

How to generate features? #15

Open Edwardmark opened 3 years ago

Edwardmark commented 3 years ago

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.