mirror of
https://github.com/pjreddie/darknet.git
synced 2023-08-10 21:13:14 +03:00
Training on VOC
This commit is contained in:
@ -4,6 +4,25 @@
|
||||
#include <stdio.h>
|
||||
#include <string.h>
|
||||
|
||||
char *get_activation_string(ACTIVATION a)
|
||||
{
|
||||
switch(a){
|
||||
case SIGMOID:
|
||||
return "sigmoid";
|
||||
case RELU:
|
||||
return "relu";
|
||||
case RAMP:
|
||||
return "ramp";
|
||||
case LINEAR:
|
||||
return "linear";
|
||||
case TANH:
|
||||
return "tanh";
|
||||
default:
|
||||
break;
|
||||
}
|
||||
return "relu";
|
||||
}
|
||||
|
||||
ACTIVATION get_activation(char *s)
|
||||
{
|
||||
if (strcmp(s, "sigmoid")==0) return SIGMOID;
|
||||
|
Reference in New Issue
Block a user