AlexeyAB / darknet

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

Reason for choosing the maxpool (size = 2, stride=1) in yolov3-tiny? #4351

Closed Sudhakar17 closed 5 years ago

Sudhakar17 commented 5 years ago

@AlexeyAB : Max Pool is for reducing the feature map size but if we choose stride=1, it retains the same size. I am talking about the final maxpool layer of yolov3-tiny. Is there any purpose for this operation?

https://github.com/AlexeyAB/darknet/blob/master/cfg/yolov3-tiny.cfg#L93

AlexeyAB commented 5 years ago

You are about - it gives +~2-4 mAP points: https://github.com/AlexeyAB/darknet/blob/ba6c35ac8965cd70804020fb4810133b3d32a5ad/cfg/yolov3-tiny.cfg#L93-L95

This is a smaller version of SPP-block: You can read about Spatial Pyramid Pooling: https://arxiv.org/abs/1406.4729v4

https://github.com/AlexeyAB/darknet/blob/ba6c35ac8965cd70804020fb4810133b3d32a5ad/cfg/yolov3-spp.cfg#L575-L597

52151356-e5d4a380-2683-11e9-9d7d-ac7bc192c477