search

pzzhang / VinVL

project page for VinVL
349 stars 25 forks source link

Pretrained model URL doesn't work. #32

Open doda999 opened 1 year ago

doda999 commented 1 year ago

It seems that this is not working now. Can you explain how to download the pretrained model?

jinz-utexas commented 1 year ago

The preextracted features are not available either.

tt0059 commented 1 year ago

Can anyone solve this problem? This is preventing researchers from continuing their research.

complete-dope commented 8 months ago

@tt0059 @doda999 have you solved this problem ? @pzzhang Any solutions for this ?

pzzhang commented 8 months ago

The checkpoint download link was updated:

pretrained models at https://drive.google.com/file/d/1nvu8y4zZFbJqSqLdQClvMyzsbBeX-oaQ/view?usp=sharing

the associated labelmap at https://drive.google.com/file/d/1M1nPtMPHS1GXx5HvKMMDxOksxQgwX14_/view?usp=sharing

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