Support yolov4

Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-04-28 20:35:30 +02:00
parent a874fad2bd
commit db0f8a4d99
15 changed files with 1979 additions and 9 deletions
@@ -48,6 +48,10 @@ dnnType* Activation::infer(dataDim_t &dim, dnnType* srcData) {
    if(act_mode == ACTIVATION_LEAKY) {
        activationLEAKYForward(srcData, dstData, dim.tot());

+    }
+    else if(act_mode == ACTIVATION_MISH) {
+        activationMishForward(srcData, dstData, dim.tot());
+
    } else {
        dnnType alpha = dnnType(1);
        dnnType beta  = dnnType(0);