AlexeyAB / darknet

YOLOv4 / Scaled-YOLOv4 / YOLO - Neural Networks for Object Detection (Windows and Linux version of Darknet )
http://pjreddie.com/darknet/
Other
21.65k stars 7.96k forks source link

Training on thermal images #7763

Open varungupta31 opened 3 years ago

varungupta31 commented 3 years ago

I have thermal image (colored) of size - Height 240 and width 320.

I'm planning to train a custom detector,

My thermal images contain a color pallet such as this:

Screenshot 2021-06-01 005350

Let me know how shall I optimize my config file that increase my chances of accurate results?

Thanks.

stephanecharette commented 3 years ago

If your images are 240x320, why not use 256x320 as your network dimensions? Or 224x320 is just as good, and slightly lower resolution so less pixels to process (which is good unless you are trying to find something small).

As for hue, sat, etc, it depends on the range you'd typically see in your images, and what you have available to train.

As for what to do about sizing...what are you looking to find? What is the size of the object compared to the size of the image? Do your training images match the images you'll be using for inference? That is what actually matters.

varungupta31 commented 3 years ago

unless you are trying to find something small

So the object i'm trying to detect would be around 20x20 or 25x20 (something in this range), minimum dimension of detection would be around 10x15. when the image is 240x320.

Also, I'd not say that detection is going to very easy, because there quite a few images where the object of interest gets a bit blurry or get mixed with the background. (if this helps in recommending a right height x width value)

Also, would 416x416 (default) be an overkill? Besides adding complexity, does it pose other disadvantages?

As for hue, sat, etc, it depends on the range you'd typically see in your images

How do I check that so I can make a better guess at setting these params (hue, etc.) if needed at all. (range in terms of color, right? the color is quite similar to image I shared in the post)

Do your training images match the images you'll be using for inference

It does.

@stephanecharette Kindly let me know, Thank you so much for the help!