chanchanchan97 / ICAFusion

ICAFusion: Iterative Cross-Attention Guided Feature Fusion for Multispectral Object Detection, Pattern Recognition
GNU Affero General Public License v3.0
118 stars 7 forks source link

The reproducibility metrics do not match the results in the paper #44

Open 3232731490 opened 1 month ago

3232731490 commented 1 month ago

When I replicated the experiment on A40, the indicator did not reach the effect in the author's paper, and the map50 was only about 0.75, did the author use the pre-training weight of yolov5-l when running the experiment?

3232731490 commented 1 month ago

When I replicated the experiment on A40, the indicator did not reach the effect in the author's paper, and the map50 was only about 0.75, did the author use the pre-training weight of yolov5-l when running the experiment?

After loading the Yolov5L pre-training weights and then training, the MAP50 indicator directly reaches about 0.81, which is higher than the index in the paper.

buyuxixi commented 2 weeks ago

请问你用的是这个配置吗? parser.add_argument('--weights', type=str, default='/root/autodl-tmp/ICAF\models\yolov5l.yaml', help='initial weights path') parser.add_argument('--cfg', type=str, default='./models/transformer/yolov5l_Transfusion_FLIR.yaml', help='model.yaml path') parser.add_argument('--data', type=str, default='/root/autodl-tmp/ICAF/data/multispectral/FLIR-align-3class.yaml', help='data.yaml path') 我为啥跑不出来0.81

chanchanchan97 commented 1 week ago

When I replicated the experiment on A40, the indicator did not reach the effect in the author's paper, and the map50 was only about 0.75, did the author use the pre-training weight of yolov5-l when running the experiment?

Yes, we use the pre-training weight during the training process.