search

chanchanchan97 / ICAFusion

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

Pre-training weight #22

Open love-whut opened 7 months ago

love-whut commented 7 months ago

Hello, did you load the pre-training weight of yolov5 during training

chanchanchan97 commented 6 months ago

Line 96-99 in train.py.

love-whut commented 6 months ago

Thank you for replying, you may have misunderstood me. I mean, do you need to load the official pre-training weight of yolov5 when training ?