Res_unit in CSP, activation function is mish not leaky

hunglc007 / tensorflow-yolov4-tflite

YOLOv4, YOLOv4-tiny, YOLOv3, YOLOv3-tiny Implemented in Tensorflow 2.0, Android. Convert YOLO v4 .weights tensorflow, tensorrt and tflite

https://github.com/hunglc007/tensorflow-yolov4-tflite

MIT License

2.23k stars 1.24k forks source link

Res_unit in CSP, activation function is mish not leaky #337

Open yingtina opened 3 years ago

yingtina commented 3 years ago

The activation function of res_unit (residual_block) In darknet from AlexeyAB's repo is mish, but in this repo is leaky. Is it right here to use leaky?

minhna1112 commented 3 years ago

You can take a look at this backbone.py file

From line 39-105 is the back bone code for CSPDarknet53, and 'mish' is used for both convolutional blocks and residual blocks

yingtina commented 3 years ago

@minhna1112 thanks!! I just checked Darknet53 not CSPDarknet53.