search

cv516Buaa / IR-VG

9 stars 0 forks source link

name 'keypoint_feature' is not defined #2

Open SudaYDD opened 3 months ago

SudaYDD commented 3 months ago

name 'keypoint_feature' is not defined File "/mnt/ssd/wyd/IR-VG-main/models/decoder.py", line 287, in forward return torch.cat([orig_img_feat, fuse_img_feat], dim=-1), text_embedding_attn, keypoint_feature File "/mnt/ssd/wyd/IR-VG-main/models/decoder.py", line 119, in forward img_feat, word_feat, kp_feature = layer(img_feat, each_feat, each_mask, img_key_padding_mask, pos, File "/mnt/ssd/wyd/IR-VG-main/models/decoder.py", line 30, in forward hs, kp = self.decoder(img_feat, each_feat, each_mask, mask, pos_embed, File "/mnt/ssd/wyd/IR-VG-main/models/VLTVG.py", line 102, in forward hs, kp = self.trans_decoder(img_feat, each_feat, word_mask_0, mask, pos_embed, word_feat, word_mask) File "/mnt/ssd/wyd/IR-VG-main/engine.py", line 83, in train_one_epoch outputs = model(img, mask, word_id, word_mask, 0, target_dict, postprocessor, target_dict['keys'], target_dict['keys_mask']) File "/mnt/ssd/wyd/IR-VG-main/train.py", line 299, in main train_stats = epoch_trainer( File "/mnt/ssd/wyd/IR-VG-main/train.py", line 414, in main(args) NameError: name 'keypoint_feature' is not defined

SudaYDD commented 3 months ago

Thank you for your awesome work! I want to ask whether there are any mistakes in the code with this version, as shown above.