search

yjh0410 / CSPDarkNet53

CSPDarkNet53
3 stars 1 forks source link

The activation function #1

Open zxm111222333 opened 2 years ago

zxm111222333 commented 2 years ago

YOLOV4 use Mish activation function, but in your code ,you use LeakyReLU activation function. Is that OK?

yjh0410 commented 2 years ago

@zxm111222333 The original YOLOv4 uses mish as the activation function. I tried, but failed. I didn't get the better performance than DarkNet-53 on ImageNet. So I just train my CSPDarkNet-53 with leakyrelu. To be honest, my CSPDarkNet-53 with LeakyReLU is not good ...

zxm111222333 commented 2 years ago

@zxm111222333 The original YOLOv4 uses mish as the activation function. I tried, but failed. I didn't get the better performance than DarkNet-53 on ImageNet. So I just train my CSPDarkNet-53 with leakyrelu. To be honest, my CSPDarkNet-53 with LeakyReLU is not good ...

OK, Thans very much for your answer.