search

AlexeyAB / darknet

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

Pretrained weights for x-ray like images #4530

Open AvaniPitre opened 4 years ago

AvaniPitre commented 4 years ago

Hi @AlexeyAB

I am working with x- ray like transparent dental images , I would like know which yolov3 pretrained weights and cfg will be sutaible for such kind of data set and any other cfg parameters need to be set before training.

Please guide

Thanks in advance Avani Pitre

AlexeyAB commented 4 years ago

@AvaniPitre Hi,

It seems that you use gray-scale images.

You can use any cfg file. For example https://raw.githubusercontent.com/AlexeyAB/darknet/master/cfg/yolov3-spp.cfg and http://pjreddie.com/media/files/darknet53.conv.74

Just set hue=0 saturation=0

If you set channels=1 in cfg-file, then you can't use pre-trained weights darknet53.conv.74, so this isn't recommended.

AvaniPitre commented 4 years ago

@AlexeyAB Thank you so much for feedback. Yes images are like gray scale but with transparent properties, is it recommended to set channel =1 ?

Thank you

AlexeyAB commented 4 years ago

is it recommended to set channel =1 ?

No.

AvaniPitre commented 4 years ago

ok thank you