added yolo4x and yolo4-csp from the github repo and download file corrections

This commit is contained in:
hchandirasekar
2021-03-26 12:18:46 +05:30
parent f3d1591430
commit f12ec3c935
12 changed files with 1444 additions and 40 deletions
+1
View File
@@ -187,6 +187,7 @@ namespace tk { namespace dnn {
if(f.activation == "relu") act = tkdnnActivationMode_t(CUDNN_ACTIVATION_RELU);
else if(f.activation == "leaky") act = tk::dnn::ACTIVATION_LEAKY;
else if(f.activation == "mish") act = tk::dnn::ACTIVATION_MISH;
else if(f.activation == "logistic") act = tk::dnn::ACTIVATION_LOGISTIC;
else { FatalError("activation not supported: " + f.activation); }
netLayers[netLayers.size()-1] = new tk::dnn::Activation(net, act);
};