|
|
|
@ -32,7 +32,7 @@ OS := $(shell uname) |
|
|
|
# GTX 1080, GTX 1070, GTX 1060, GTX 1050, GTX 1030, Titan Xp, Tesla P40, Tesla P4
|
|
|
|
# GTX 1080, GTX 1070, GTX 1060, GTX 1050, GTX 1030, Titan Xp, Tesla P40, Tesla P4
|
|
|
|
# ARCH= -gencode arch=compute_61,code=sm_61 -gencode arch=compute_61,code=compute_61
|
|
|
|
# ARCH= -gencode arch=compute_61,code=sm_61 -gencode arch=compute_61,code=compute_61
|
|
|
|
|
|
|
|
|
|
|
|
# GP100/Tesla P100 – DGX-1
|
|
|
|
# GP100/Tesla P100 - DGX-1
|
|
|
|
# ARCH= -gencode arch=compute_60,code=sm_60
|
|
|
|
# ARCH= -gencode arch=compute_60,code=sm_60
|
|
|
|
|
|
|
|
|
|
|
|
# For Jetson TX1, Tegra X1, DRIVE CX, DRIVE PX - uncomment:
|
|
|
|
# For Jetson TX1, Tegra X1, DRIVE CX, DRIVE PX - uncomment:
|
|
|
|
@ -53,30 +53,30 @@ endif |
|
|
|
|
|
|
|
|
|
|
|
CC=gcc
|
|
|
|
CC=gcc
|
|
|
|
CPP=g++
|
|
|
|
CPP=g++
|
|
|
|
NVCC=nvcc
|
|
|
|
NVCC=nvcc
|
|
|
|
OPTS=-Ofast
|
|
|
|
OPTS=-Ofast
|
|
|
|
LDFLAGS= -lm -pthread
|
|
|
|
LDFLAGS= -lm -pthread
|
|
|
|
COMMON= -Iinclude/
|
|
|
|
COMMON= -Iinclude/ -I3rdparty/stb/include
|
|
|
|
CFLAGS=-Wall -Wfatal-errors -Wno-unused-result -Wno-unknown-pragmas -fPIC
|
|
|
|
CFLAGS=-Wall -Wfatal-errors -Wno-unused-result -Wno-unknown-pragmas -fPIC
|
|
|
|
|
|
|
|
|
|
|
|
ifeq ($(DEBUG), 1)
|
|
|
|
ifeq ($(DEBUG), 1) |
|
|
|
#OPTS= -O0 -g
|
|
|
|
#OPTS= -O0 -g
|
|
|
|
#OPTS= -Og -g
|
|
|
|
#OPTS= -Og -g
|
|
|
|
COMMON+= -DDEBUG
|
|
|
|
COMMON+= -DDEBUG
|
|
|
|
CFLAGS+= -DDEBUG
|
|
|
|
CFLAGS+= -DDEBUG
|
|
|
|
else |
|
|
|
else |
|
|
|
ifeq ($(AVX), 1)
|
|
|
|
ifeq ($(AVX), 1) |
|
|
|
CFLAGS+= -ffp-contract=fast -mavx -mavx2 -msse3 -msse4.1 -msse4.2 -msse4a
|
|
|
|
CFLAGS+= -ffp-contract=fast -mavx -mavx2 -msse3 -msse4.1 -msse4.2 -msse4a
|
|
|
|
endif |
|
|
|
endif |
|
|
|
endif |
|
|
|
endif |
|
|
|
|
|
|
|
|
|
|
|
CFLAGS+=$(OPTS)
|
|
|
|
CFLAGS+=$(OPTS)
|
|
|
|
|
|
|
|
|
|
|
|
ifeq ($(OPENCV), 1)
|
|
|
|
ifeq ($(OPENCV), 1) |
|
|
|
COMMON+= -DOPENCV
|
|
|
|
COMMON+= -DOPENCV
|
|
|
|
CFLAGS+= -DOPENCV
|
|
|
|
CFLAGS+= -DOPENCV
|
|
|
|
LDFLAGS+= `pkg-config --libs opencv`
|
|
|
|
LDFLAGS+= `pkg-config --libs opencv`
|
|
|
|
COMMON+= `pkg-config --cflags opencv`
|
|
|
|
COMMON+= `pkg-config --cflags opencv`
|
|
|
|
endif |
|
|
|
endif |
|
|
|
|
|
|
|
|
|
|
|
ifeq ($(OPENMP), 1) |
|
|
|
ifeq ($(OPENMP), 1) |
|
|
|
@ -112,8 +112,8 @@ ARCH+= -gencode arch=compute_70,code=[sm_70,compute_70] |
|
|
|
endif |
|
|
|
endif |
|
|
|
|
|
|
|
|
|
|
|
OBJ=http_stream.o gemm.o utils.o cuda.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 darknet.o detection_layer.o captcha.o route_layer.o writing.o box.o nightmare.o normalization_layer.o avgpool_layer.o coco.o dice.o yolo.o detector.o layer.o compare.o classifier.o local_layer.o swag.o shortcut_layer.o activation_layer.o rnn_layer.o gru_layer.o rnn.o rnn_vid.o crnn_layer.o demo.o tag.o cifar.o go.o batchnorm_layer.o art.o region_layer.o reorg_layer.o reorg_old_layer.o super.o voxel.o tree.o yolo_layer.o upsample_layer.o lstm_layer.o
|
|
|
|
OBJ=http_stream.o gemm.o utils.o cuda.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 darknet.o detection_layer.o captcha.o route_layer.o writing.o box.o nightmare.o normalization_layer.o avgpool_layer.o coco.o dice.o yolo.o detector.o layer.o compare.o classifier.o local_layer.o swag.o shortcut_layer.o activation_layer.o rnn_layer.o gru_layer.o rnn.o rnn_vid.o crnn_layer.o demo.o tag.o cifar.o go.o batchnorm_layer.o art.o region_layer.o reorg_layer.o reorg_old_layer.o super.o voxel.o tree.o yolo_layer.o upsample_layer.o lstm_layer.o
|
|
|
|
ifeq ($(GPU), 1)
|
|
|
|
ifeq ($(GPU), 1) |
|
|
|
LDFLAGS+= -lstdc++
|
|
|
|
LDFLAGS+= -lstdc++
|
|
|
|
OBJ+=convolutional_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 network_kernels.o avgpool_layer_kernels.o
|
|
|
|
OBJ+=convolutional_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 network_kernels.o avgpool_layer_kernels.o
|
|
|
|
endif |
|
|
|
endif |
|
|
|
|
|
|
|
|
|
|
|
@ -122,12 +122,12 @@ DEPS = $(wildcard src/*.h) Makefile include/darknet.h |
|
|
|
|
|
|
|
|
|
|
|
all: obj backup results setchmod $(EXEC) $(LIBNAMESO) $(APPNAMESO) |
|
|
|
all: obj backup results setchmod $(EXEC) $(LIBNAMESO) $(APPNAMESO) |
|
|
|
|
|
|
|
|
|
|
|
ifeq ($(LIBSO), 1)
|
|
|
|
ifeq ($(LIBSO), 1) |
|
|
|
CFLAGS+= -fPIC
|
|
|
|
CFLAGS+= -fPIC
|
|
|
|
|
|
|
|
|
|
|
|
$(LIBNAMESO): $(OBJS) include/yolo_v2_class.hpp src/yolo_v2_class.cpp |
|
|
|
$(LIBNAMESO): $(OBJS) include/yolo_v2_class.hpp src/yolo_v2_class.cpp |
|
|
|
$(CPP) -shared -std=c++11 -fvisibility=hidden -DLIB_EXPORTS $(COMMON) $(CFLAGS) $(OBJS) src/yolo_v2_class.cpp -o $@ $(LDFLAGS)
|
|
|
|
$(CPP) -shared -std=c++11 -fvisibility=hidden -DLIB_EXPORTS $(COMMON) $(CFLAGS) $(OBJS) src/yolo_v2_class.cpp -o $@ $(LDFLAGS)
|
|
|
|
|
|
|
|
|
|
|
|
$(APPNAMESO): $(LIBNAMESO) include/yolo_v2_class.hpp src/yolo_console_dll.cpp |
|
|
|
$(APPNAMESO): $(LIBNAMESO) include/yolo_v2_class.hpp src/yolo_console_dll.cpp |
|
|
|
$(CPP) -std=c++11 $(COMMON) $(CFLAGS) -o $@ src/yolo_console_dll.cpp $(LDFLAGS) -L ./ -l:$(LIBNAMESO)
|
|
|
|
$(CPP) -std=c++11 $(COMMON) $(CFLAGS) -o $@ src/yolo_console_dll.cpp $(LDFLAGS) -L ./ -l:$(LIBNAMESO)
|
|
|
|
endif |
|
|
|
endif |
|
|
|
@ -157,4 +157,3 @@ setchmod: |
|
|
|
|
|
|
|
|
|
|
|
clean: |
|
|
|
clean: |
|
|
|
rm -rf $(OBJS) $(EXEC) $(LIBNAMESO) $(APPNAMESO)
|
|
|
|
rm -rf $(OBJS) $(EXEC) $(LIBNAMESO) $(APPNAMESO)
|
|
|
|
|
|
|
|
|
|
|
|
|
Stefano Sinigardi
6 years ago