-
Notifications
You must be signed in to change notification settings - Fork 3
/
main.c
36 lines (28 loc) · 1.05 KB
/
main.c
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
#include "darknet.h"
#include <stdio.h>
// 函数的声明中带有关键字extern, 仅仅是暗示这个函数可能在别的源文档里定义.
extern void run_classifier(int argc, char **argv);
int main(int argc, char **argv)
{
// /************下面配置用于分类网络的代码分析****************************
// 参考: https://pjreddie.com/darknet/train-cifar/
argc = 5;
char* parameters[] = {"./darknet",
"classifier",
"train",
"/home/cxj/Desktop/code/darknet-mini/cfg/data/cifar.data",
"/home/cxj/Desktop/code/darknet-mini/cfg/net/cls_cifar_small.cfg"};
argv = parameters;
// *****************************************************************/
if(argc < 2){
fprintf(stderr, "usage: %s <function>\n", argv[0]);
return 0;
}
if (0 == strcmp(argv[1], "classifier")){
run_classifier(argc, argv);
}
else {
fprintf(stderr, "Not an option: %s\n", argv[1]);
}
return 0;
}