search

rentainhe / TRAR-VQA

[ICCV 2021] Official implementation of the paper "TRAR: Routing the Attention Spans in Transformers for Visual Question Answering"
MIT License
64 stars 18 forks source link

16*16 #7

Open erjpc opened 1 month ago

erjpc commented 1 month ago

Hello author, when I tried to extract 1616 features python extract_trar_grid_feature.py \ --config-file configs/X-152-grid.yaml \ --dataset coco_2014_train \ --output_dir /data_home/jiangpeichao/data/vqa_Grid_feature16/feats \ --weight_path /data_home/jiangpeichao/X-152.pth \ --feature_size 16 An error occurred： Traceback (most recent call last): File "extract_trar_grid_feature.py", line 120, in main(args) File "extract_trar_grid_feature.py", line 105, in main model = build_model(cfg) File "/data_home/jiangpeichao/detectron2/detectron2/modeling/meta_arch/build.py", line 22, in build_model model = META_ARCH_REGISTRY.get(meta_arch)(cfg) File "/data_home/jiangpeichao/detectron2/detectron2/config/config.py", line 189, in wrapped explicit_args = _get_args_from_config(from_config_func, args, *kwargs) File "/data_home/jiangpeichao/detectron2/detectron2/config/config.py", line 245, in _get_args_from_config ret = from_config_func(args, kwargs) File "/data_home/jiangpeichao/detectron2/detectron2/modeling/meta_arch/rcnn.py", line 77, in from_config "roi_heads": build_roi_heads(cfg, backbone.output_shape()), File "/data_home/jiangpeichao/detectron2/detectron2/modeling/roi_heads/roi_heads.py", line 43, in build_roi_heads return ROI_HEADS_REGISTRY.get(name)(cfg, input_shape) File "/data_home/jiangpeichao/TRAR-Feature-Extraction-master/grid_feats/roi_heads.py", line 175, in init super(StandardROIHeads, self).init(cfg, input_shape) File "/data_home/jiangpeichao/detectron2/detectron2/config/config.py", line 190, in wrapped init_func(self, explicit_args) TypeError: init() got an unexpected keyword argument 'train_on_pred_boxes' Can you help me with it