AttributeError: 'BertModel' 对象没有属性 'get_head_mask'
在运行 shell 脚本 CUDA_VISIBLE_DEVICES=0 Python demo/inference_on_a_image.py -c groundingdino/config/GroundingDINO_SwinT_OGC.py -p weights/groundingdino_swint_ogc.pth -i /root/GroundingDINO/data/VOC-YOLO/images/test/000001.jpg -o "out" -t "dog" 时,出现了错误:Traceback (most recent call last): File "/root/GroundingDINO/demo/inference_on_a_image.py", line 189, in model = load_model(config_file, checkpoint_path, cpu_only=args.cpu_only) File "/root/GroundingDINO/demo/inference_on_a_image.py", line 76, in load_model model = build_model(args) File "/root/GroundingDINO/groundingdino/models/init.py", line 17, in build_model model = build_func(args) File "/root/GroundingDINO/groundingdino/models/GroundingDINO/groundingdino.py", line 388, in build_groundingdino model = GroundingDINO( File "/root/GroundingDINO/groundingdino/models/GroundingDINO/groundingdino.py", line 111, in init self.bert = BertModelWarper(bert_model=self.bert) File "/root/GroundingDINO/groundingdino/models/GroundingDINO/bertwarper.py", line 29, in init self.get_head_mask = bert_model.get_head_mask File "/usr/local/miniconda3/envs/dino/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1928, in getattr raise AttributeError('BertModel' object has no attribute 'get_head_mask')`
在搜索互联网后,我意识到这个错误可能是由于最近更新的包 Transformer 所引起的。
内容来源: IDEA-Research/GroundingDINO