AlexeyAB / darknet

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

questions about new release #4526

Open tuteming opened 4 years ago

tuteming commented 4 years ago

what function are Stretch conv-kernel, Rotate conv-kernel, and Sway conv-kernel? can give a corresponding issue or refs?

hoangphucITJP commented 4 years ago

I want to know about Rotate conv-kernel also (the commit https://github.com/AlexeyAB/darknet/commit/5e035565ff6226aaea3377bce9bb8e2279f828ae). Does it mean that the model is rotation-invariant?

AlexeyAB commented 4 years ago

@tuteming @hoangphucITJP

This is experimental training approach. Yes, it tries to make conv layers rotation-invariant and scale-invariant

should be used as

[convolutional]
batch_normalize=1
filters=1024     # 1024
groups=4
size=3
stride=1
pad=1
activation=leaky
rotate=1

[maxpool]
maxpool_depth=1
out_channels=256       # 1024 /4 = 256
size=1
stride=1

You can look at this table - field mAP@0.5 valid-set (other videos) : https://github.com/AlexeyAB/darknet/issues/4495#issuecomment-578538967

So you can try to use:

And train as Yolov3-tiny model: https://github.com/AlexeyAB/darknet#how-to-train-tiny-yolo-to-detect-your-custom-objects

./darknet detector train data/obj.data yolov3-tiny-obj.cfg yolov3-tiny.conv.15