YOLOonMe / EMA-attention-module

Implementation Code for the ICCASSP 2023 paper " Efficient Multi-Scale Attention Module with Cross-Spatial Learning" and is available at: https://arxiv.org/abs/2305.13563v2
163 stars 9 forks source link

pre-trained weights #7

Open LeiaJ520 opened 9 months ago

LeiaJ520 commented 9 months ago

Hi, by inserting the attention module into backbone, will the original yolov5 pre-trained weights still be available? I think that if the backbone structure is modified, the corresponding parameters will also change, and the weights that have been learned can not be used, and we have to start training from scratch. I don't know if my understanding is correct, and I ask the author for an answer.