forked from pjreddie/darknet
-
Notifications
You must be signed in to change notification settings - Fork 5
/
Makefile
155 lines (127 loc) · 4.43 KB
/
Makefile
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
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
GPU=0
CUDNN=0
OPENCV=0
OPENMP=0
DEBUG=0
CAFFE=0
ODLA=0
ARCH= -gencode arch=compute_30,code=sm_30 \
-gencode arch=compute_35,code=sm_35 \
-gencode arch=compute_50,code=[sm_50,compute_50] \
-gencode arch=compute_52,code=[sm_52,compute_52]
# -gencode arch=compute_20,code=[sm_20,sm_21] \ This one is deprecated?
# This is what I use, uncomment if you know your arch and want to specify
# ARCH= -gencode arch=compute_52,code=compute_52
VPATH=./src/:./examples
ifeq ($(CAFFE), 1)
ifeq ($(ODLA), 1)
$(error Can't build both ODLA and Caffe at the same time)
endif
CAFFELIB=libcaffelayer.so
CAFFEOBJ=obj/caffe_layer_impl.o
ALIB=
else
CAFFELIB=
CAFFEOBJ=
ALIB=libdarknet.a
endif
ifeq ($(ODLA), 1)
ODLALIB=libodlalayer.so
ODLAOBJ=obj/odla_layer_impl.o
ALIB=
else
ODLALIB=
ODLAOBJ=
ALIB=libdarknet.a
endif
SLIB=libdarknet.so
EXEC=darknet
OBJDIR=./obj/
ifeq ($(ODLA), 1)
CPP=/home/secondary/OpenDLA/buildroot/output/host/opt/ext-toolchain/bin/aarch64-linux-gnu-g++
CC=/home/secondary/OpenDLA/buildroot/output/host/opt/ext-toolchain/bin/aarch64-linux-gnu-gcc
AR=/home/secondary/OpenDLA/buildroot/output/host/opt/ext-toolchain/bin/aarch64-linux-gnu-ar
else
CPP=g++
CC=gcc
AR=ar
endif
NVCC=nvcc
ARFLAGS=rcs
OPTS=-Ofast
LDFLAGS= -lm -pthread
COMMON= -Iinclude/ -Isrc/
CFLAGS=-Wall -Wno-unused-result -Wno-unknown-pragmas -Wfatal-errors -fPIC
ifeq ($(OPENMP), 1)
CFLAGS+= -fopenmp
endif
ifeq ($(DEBUG), 1)
OPTS=-O0 -g
endif
CFLAGS+=$(OPTS)
ifeq ($(OPENCV), 1)
COMMON+= -DOPENCV
CFLAGS+= -DOPENCV
LDFLAGS+= `pkg-config --libs opencv`
COMMON+= `pkg-config --cflags opencv`
endif
ifeq ($(GPU), 1)
COMMON+= -DGPU -I/usr/local/cuda/include/
CFLAGS+= -DGPU
LDFLAGS+= -L/usr/local/cuda/lib64 -lcuda -lcudart -lcublas -lcurand
endif
ifeq ($(CUDNN), 1)
COMMON+= -DCUDNN
CFLAGS+= -DCUDNN
LDFLAGS+= -lcudnn
endif
OBJ=gemm.o utils.o cuda.o deconvolutional_layer.o convolutional_layer.o list.o image.o activations.o im2col.o col2im.o blas.o crop_layer.o dropout_layer.o maxpool_layer.o softmax_layer.o data.o matrix.o network.o connected_layer.o cost_layer.o parser.o option_list.o detection_layer.o route_layer.o upsample_layer.o box.o normalization_layer.o avgpool_layer.o layer.o local_layer.o shortcut_layer.o logistic_layer.o activation_layer.o rnn_layer.o gru_layer.o crnn_layer.o demo.o batchnorm_layer.o region_layer.o reorg_layer.o tree.o lstm_layer.o l2norm_layer.o yolo_layer.o odla_layer.o caffe_layer.o split_layer.o upsample_odla_layer.o converter.o converter_layer.o
EXECOBJA=captcha.o lsd.o super.o art.o tag.o cifar.o go.o rnn.o segmenter.o regressor.o classifier.o coco.o yolo.o detector.o nightmare.o darknet.o
ifeq ($(GPU), 1)
LDFLAGS+= -lstdc++
OBJ+=convolutional_kernels.o deconvolutional_kernels.o activation_kernels.o im2col_kernels.o col2im_kernels.o blas_kernels.o crop_layer_kernels.o dropout_layer_kernels.o maxpool_layer_kernels.o avgpool_layer_kernels.o
endif
ifeq ($(CAFFE), 1)
LDFLAGS+= -L/usr/lib -L/home/secondary/OpenDLA/YOLO/share/darknet -lcaffelayer -lcaffe
LNKDARKNET= -ldarknet
endif
ifeq ($(ODLA), 1)
LDFLAGS+= -L/home/secondary/OpenDLA/YOLO/share/darknet -lnvdla_runtime
LNKDARKNET= -ldarknet
endif
EXECOBJ = $(addprefix $(OBJDIR), $(EXECOBJA))
OBJS = $(addprefix $(OBJDIR), $(OBJ))
DEPS = $(wildcard src/*.h) Makefile include/darknet.h
all: obj backup results $(ODLALIB) $(CAFFELIB) $(SLIB) $(ALIB) $(EXEC)
#all: obj results $(SLIB) $(ALIB) $(EXEC)
$(EXEC): $(EXECOBJ) $(ODLALIB) $(CAFFELIB) $(SLIB) $(ALIB)
$(CC) $(COMMON) $(CFLAGS) $^ -o $@ $(LDFLAGS) $(LNKDARKNET) $(ALIB)
$(ALIB): $(OBJS)
$(AR) $(ARFLAGS) $@ $^
$(SLIB): $(OBJS)
$(CC) $(CFLAGS) -shared $^ -o $@ $(LDFLAGS)
ifeq ($(CAFFE), 1)
$(CAFFEOBJ):
$(CPP) -I/usr/include -Wall -fPIC -O -c src/caffe_layer_impl.cpp -o obj/caffe_layer_impl.o
$(CAFFELIB): $(CAFFEOBJ)
$(CPP) -shared obj/caffe_layer_impl.o -o libcaffelayer.so -L/usr/lib -L/home/secondary/OpenDLA/YOLO/darknet -lcaffe
endif
ifeq ($(ODLA), 1)
$(ODLAOBJ):
$(CPP) -Wall -fPIC -O -c src/odla_layer_impl.cpp -o obj/odla_layer_impl.o
$(ODLALIB): $(ODLAOBJ)
$(CPP) -shared obj/odla_layer_impl.o -o libodlalayer.so -L/home/secondary/OpenDLA/YOLO/darknet -lnvdla_runtime
endif
$(OBJDIR)%.o: %.c $(DEPS)
$(CC) $(COMMON) $(CFLAGS) -c $< -o $@
$(OBJDIR)%.o: %.cu $(DEPS)
$(NVCC) $(ARCH) $(COMMON) --compiler-options "$(CFLAGS)" -c $< -o $@
obj:
mkdir -p obj
backup:
mkdir -p backup
results:
mkdir -p results
.PHONY: clean
clean:
rm -rf $(ODLAOBJ) $(ODLALIB) $(CAFFEOBJ) $(CAFFELIB) $(OBJS) $(SLIB) $(ALIB) $(EXEC) $(EXECOBJ) $(OBJDIR)/*