DocF / multispectral-object-detection

Multispectral Object Detection with Yolov5 and Transformer
GNU Affero General Public License v3.0
315 stars 60 forks source link

dataset #12

Closed 377fsq closed 2 years ago

377fsq commented 2 years ago

Can you provide image file structure of LLVIP?

DocF commented 2 years ago

You need to convert all annotations to YOLOv5 format.

Refer: https://github.com/ultralytics/yolov5/wiki/Train-Custom-Data

377fsq commented 2 years ago

Thanks for your answer, I have finished the training. But there are some questions. I used the yolov5l_fusion_transformerx3_llvip.yaml to train with default parameters, but the training results did not reach the values shown in your paper. My result mAP50: 0.965 mAP75: 0.709 mAP: 0.624 GPU: 3090Ti batch_size: 16. I want to know whether add other training strategies?

DocF commented 2 years ago

First, my input image size is 1024×1024. Second, I tested thebest.pt.

377fsq commented 2 years ago

Thank you, I will try again.

nineRishav commented 1 year ago

Hi @377fsq ,

I need you help in running the project. I wanted to train, test the model. But, I am facing issue in mentioning the path of FLIR_alligned. image

What should I enter in the 4 fields of train_rgb, val_rgb, train_ir, val_ir.