search

AlexeyAB / darknet

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

Tinier Tiny Yolo v3? #6727

Open GotG opened 4 years ago

GotG commented 4 years ago

Hello

I am looking for a tiny yolo v3 that has less parameters (by 1/3 or so), so that it is at least 20% faster on inference. Looking at this paper: https://ieeexplore.ieee.org/document/8941141

Does anyone have experience with this Tinier version, or smth similar that would match what I am looking for?

Thanks

stephanecharette commented 4 years ago

Definitely post back if you implement it, or if you find a copy of the .cfg file. I'd also be interested in a tinier/faster config.

eyebies commented 4 years ago

https://github.com/jnfrancis/Tinier-YOLO by first author

scamianbas commented 4 years ago

Training with Tinier-1119-1.cfg and current @AlexeyAB Yolo works like a charm ^^

eyebies commented 4 years ago

Is it any good?

scamianbas commented 4 years ago

chart_Tinier-1119-1

Not so bad, about 75 FPS, needs to be trained a bit more

GotG commented 4 years ago

@scamianbas did you make any changes to the config? i used this (low number of batches to test the model) learning_rate=0.0005 max_batches = 5000 policy=steps steps=4000,4500 scales=.1,.1 burn_in=1000 chart_tinierv3

scamianbas commented 4 years ago

@GotG I only changed max_batches = 20000 and steps=16000,18000 accordingly And also, accordingly to my dataset, classes=2 and filters=21 ( (classes + 5)x3 )