search

joewybean / MMI-Det

The official version of the paper "MMI-Det: Exploring Multi-Modal Integration for Visible and Infrared Object Detection"
Apache License 2.0
2 stars 0 forks source link

how to train? #2

Open jacksonwu09 opened 2 weeks ago

jacksonwu09 commented 2 weeks ago

Hello, I would like to ask how to use this repository. Can I just run python train.py directly?

joewybean commented 2 weeks ago

Yes, my code is based on Yolov5, you can try

joewybean commented 2 weeks ago

By the way, you need to prepare the multi-spectrual Dataset like M3FD, FLIR-align, and LLVIP, split the datasets into train and val datasets.

jacksonwu09 commented 1 week ago

Okay, thanks! I might need to test it on a new dataset. I'd like to ask about the format modifications required for a new dataset. Could you give me some suggestions?

joewybean commented 3 days ago

--datasets ----M3FD ------RGBimages --------train ----------0001.png ----------0002.png --------val ----------0001.png ----------0002.png ------IRimages --------train ----------0001.png ----------0002.png --------val ----------0001.png ----------0002.png ------labels --------train ----------0001.txt ----------0002.txt --------val ----------0001.txt ----------0002.txt