mirror of
https://github.com/LdDl/go-darknet.git
synced 2025-09-26 19:51:27 +08:00
add docker container for cpu/gpu
This commit is contained in:
lucmichalski
1
docker/.env
Normal file
1
docker/.env
Normal file
@@ -0,0 +1 @@
|
||||
NAMESPACE=darknet
|
||||
40
docker/Dockerfile
Normal file
40
docker/Dockerfile
Normal file
@@ -0,0 +1,40 @@
|
||||
# Build phase
|
||||
FROM ubuntu:18.04 as builder
|
||||
|
||||
ENV darknet_commit=a234a5022333c930de08f2470184ef4e0c68356e
|
||||
|
||||
WORKDIR /root/build
|
||||
COPY Makefile.cpu .
|
||||
RUN apt-get -y update && \
|
||||
apt-get -y install --no-install-recommends git build-essential ca-certificates && \
|
||||
git clone https://github.com/AlexeyAB/darknet && \
|
||||
cd darknet && \
|
||||
git checkout $darknet_commit && \
|
||||
cp -f /root/build/Makefile.cpu Makefile && \
|
||||
make
|
||||
|
||||
# Final Image
|
||||
FROM golang:1.14
|
||||
|
||||
RUN apt-get -y update && \
|
||||
apt-get -y install --no-install-recommends nano bash jq
|
||||
|
||||
WORKDIR /root
|
||||
COPY --from=builder /root/build/darknet/darknet \
|
||||
/root/build/darknet/libdarknet.so \
|
||||
/root/build/darknet/include/darknet.h \
|
||||
./staging/
|
||||
|
||||
RUN mv staging/darknet /usr/local/bin && \
|
||||
mv staging/darknet.h /usr/include && \
|
||||
mv staging/libdarknet.so /usr/lib && \
|
||||
rm -rf staging
|
||||
|
||||
RUN go get -u github.com/LdDl/go-darknet \
|
||||
&& go get -u github.com/disintegration/imaging
|
||||
|
||||
WORKDIR /darknet
|
||||
|
||||
COPY download_data.sh .
|
||||
|
||||
CMD ["/bin/bash"]
|
||||
38
docker/Dockerfile.gpu
Normal file
38
docker/Dockerfile.gpu
Normal file
@@ -0,0 +1,38 @@
|
||||
# Build phase
|
||||
FROM nvidia/cuda:9.0-cudnn7-devel-ubuntu16.04 as builder
|
||||
|
||||
ENV darknet_commit=680d3bd
|
||||
|
||||
WORKDIR /root/build
|
||||
COPY darknet.patch .
|
||||
RUN apt-get -y update && \
|
||||
apt-get -y install git build-essential && \
|
||||
git clone https://github.com/pjreddie/darknet.git && \
|
||||
cd darknet && \
|
||||
git checkout $darknet_commit && \
|
||||
git apply ../darknet.patch && \
|
||||
make
|
||||
|
||||
# Final Image
|
||||
FROM nvidia/cuda:9.0-cudnn7-runtime-ubuntu16.04
|
||||
|
||||
WORKDIR /root
|
||||
COPY --from=builder /root/build/darknet/darknet \
|
||||
/root/build/darknet/libdarknet.so \
|
||||
/root/build/darknet/libdarknet.a \
|
||||
/root/build/darknet/include/darknet.h \
|
||||
./staging/
|
||||
|
||||
RUN mv staging/darknet /usr/local/bin && \
|
||||
mv staging/darknet.h /usr/local/include && \
|
||||
mv staging/libdarknet.so staging/libdarknet.a /usr/local/lib && \
|
||||
rm -rf staging
|
||||
|
||||
RUN go get -u github.com/LdDl/go-darknet \
|
||||
&& go get -u github.com/disintegration/imaging
|
||||
|
||||
WORKDIR /darknet
|
||||
|
||||
COPY download_data.sh .
|
||||
|
||||
CMD ["/bin/bash"]
|
||||
185
docker/Makefile.cpu
Normal file
185
docker/Makefile.cpu
Normal file
@@ -0,0 +1,185 @@
|
||||
GPU=0
|
||||
CUDNN=0
|
||||
CUDNN_HALF=0
|
||||
OPENCV=0
|
||||
AVX=1
|
||||
OPENMP=1
|
||||
LIBSO=1
|
||||
ZED_CAMERA=0 # ZED SDK 3.0 and above
|
||||
ZED_CAMERA_v2_8=0 # ZED SDK 2.X
|
||||
|
||||
# set GPU=1 and CUDNN=1 to speedup on GPU
|
||||
# set CUDNN_HALF=1 to further speedup 3 x times (Mixed-precision on Tensor Cores) GPU: Volta, Xavier, Turing and higher
|
||||
# set AVX=1 and OPENMP=1 to speedup on CPU (if error occurs then set AVX=0)
|
||||
|
||||
USE_CPP=0
|
||||
DEBUG=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_61,code=[sm_61,compute_61]
|
||||
|
||||
OS := $(shell uname)
|
||||
|
||||
# Tesla V100
|
||||
# ARCH= -gencode arch=compute_70,code=[sm_70,compute_70]
|
||||
|
||||
# GeForce RTX 2080 Ti, RTX 2080, RTX 2070, Quadro RTX 8000, Quadro RTX 6000, Quadro RTX 5000, Tesla T4, XNOR Tensor Cores
|
||||
# ARCH= -gencode arch=compute_75,code=[sm_75,compute_75]
|
||||
|
||||
# Jetson XAVIER
|
||||
# ARCH= -gencode arch=compute_72,code=[sm_72,compute_72]
|
||||
|
||||
# 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
|
||||
|
||||
# GP100/Tesla P100 - DGX-1
|
||||
# ARCH= -gencode arch=compute_60,code=sm_60
|
||||
|
||||
# For Jetson TX1, Tegra X1, DRIVE CX, DRIVE PX - uncomment:
|
||||
# ARCH= -gencode arch=compute_53,code=[sm_53,compute_53]
|
||||
|
||||
# For Jetson Tx2 or Drive-PX2 uncomment:
|
||||
# ARCH= -gencode arch=compute_62,code=[sm_62,compute_62]
|
||||
|
||||
|
||||
# VPATH=./src/
|
||||
VPATH=./src/:./examples
|
||||
SLIB=libdarknet.so
|
||||
|
||||
EXEC=darknet
|
||||
OBJDIR=./obj/
|
||||
|
||||
ifeq ($(LIBSO), 1)
|
||||
LIBNAMESO=libdarknet.so
|
||||
APPNAMESO=uselib
|
||||
endif
|
||||
|
||||
ifeq ($(USE_CPP), 1)
|
||||
CC=g++
|
||||
else
|
||||
CC=gcc
|
||||
endif
|
||||
|
||||
CPP=g++ -std=c++11
|
||||
NVCC=nvcc
|
||||
OPTS=-Ofast
|
||||
LDFLAGS= -lm -pthread
|
||||
COMMON= -Iinclude/ -I3rdparty/stb/include
|
||||
CFLAGS=-Wall -Wfatal-errors -Wno-unused-result -Wno-unknown-pragmas -fPIC
|
||||
|
||||
ifeq ($(DEBUG), 1)
|
||||
#OPTS= -O0 -g
|
||||
#OPTS= -Og -g
|
||||
COMMON+= -DDEBUG
|
||||
CFLAGS+= -DDEBUG
|
||||
else
|
||||
ifeq ($(AVX), 1)
|
||||
CFLAGS+= -ffp-contract=fast -mavx -mavx2 -msse3 -msse4.1 -msse4.2 -msse4a
|
||||
endif
|
||||
endif
|
||||
|
||||
CFLAGS+=$(OPTS)
|
||||
|
||||
ifneq (,$(findstring MSYS_NT,$(OS)))
|
||||
LDFLAGS+=-lws2_32
|
||||
endif
|
||||
|
||||
ifeq ($(OPENCV), 1)
|
||||
COMMON+= -DOPENCV
|
||||
CFLAGS+= -DOPENCV
|
||||
LDFLAGS+= `pkg-config --libs opencv4 2> /dev/null || pkg-config --libs opencv`
|
||||
COMMON+= `pkg-config --cflags opencv4 2> /dev/null || pkg-config --cflags opencv`
|
||||
endif
|
||||
|
||||
ifeq ($(OPENMP), 1)
|
||||
CFLAGS+= -fopenmp
|
||||
LDFLAGS+= -lgomp
|
||||
endif
|
||||
|
||||
ifeq ($(GPU), 1)
|
||||
COMMON+= -DGPU -I/usr/local/cuda/include/
|
||||
CFLAGS+= -DGPU
|
||||
ifeq ($(OS),Darwin) #MAC
|
||||
LDFLAGS+= -L/usr/local/cuda/lib -lcuda -lcudart -lcublas -lcurand
|
||||
else
|
||||
LDFLAGS+= -L/usr/local/cuda/lib64 -lcuda -lcudart -lcublas -lcurand
|
||||
endif
|
||||
endif
|
||||
|
||||
ifeq ($(CUDNN), 1)
|
||||
COMMON+= -DCUDNN
|
||||
ifeq ($(OS),Darwin) #MAC
|
||||
CFLAGS+= -DCUDNN -I/usr/local/cuda/include
|
||||
LDFLAGS+= -L/usr/local/cuda/lib -lcudnn
|
||||
else
|
||||
CFLAGS+= -DCUDNN -I/usr/local/cudnn/include
|
||||
LDFLAGS+= -L/usr/local/cudnn/lib64 -lcudnn
|
||||
endif
|
||||
endif
|
||||
|
||||
ifeq ($(CUDNN_HALF), 1)
|
||||
COMMON+= -DCUDNN_HALF
|
||||
CFLAGS+= -DCUDNN_HALF
|
||||
ARCH+= -gencode arch=compute_70,code=[sm_70,compute_70]
|
||||
endif
|
||||
|
||||
ifeq ($(ZED_CAMERA), 1)
|
||||
CFLAGS+= -DZED_STEREO -I/usr/local/zed/include
|
||||
ifeq ($(ZED_CAMERA_v2_8), 1)
|
||||
LDFLAGS+= -L/usr/local/zed/lib -lsl_core -lsl_input -lsl_zed
|
||||
#-lstdc++ -D_GLIBCXX_USE_CXX11_ABI=0
|
||||
else
|
||||
LDFLAGS+= -L/usr/local/zed/lib -lsl_zed
|
||||
#-lstdc++ -D_GLIBCXX_USE_CXX11_ABI=0
|
||||
endif
|
||||
endif
|
||||
|
||||
OBJ=image_opencv.o http_stream.o gemm.o utils.o dark_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 gaussian_yolo_layer.o upsample_layer.o lstm_layer.o conv_lstm_layer.o scale_channels_layer.o sam_layer.o
|
||||
ifeq ($(GPU), 1)
|
||||
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
|
||||
endif
|
||||
|
||||
OBJS = $(addprefix $(OBJDIR), $(OBJ))
|
||||
DEPS = $(wildcard src/*.h) Makefile include/darknet.h
|
||||
|
||||
all: $(OBJDIR) backup results setchmod $(EXEC) $(LIBNAMESO) $(APPNAMESO)
|
||||
|
||||
ifeq ($(LIBSO), 1)
|
||||
CFLAGS+= -fPIC
|
||||
|
||||
$(LIBNAMESO): $(OBJDIR) $(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)
|
||||
|
||||
$(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)
|
||||
endif
|
||||
|
||||
$(EXEC): $(OBJS)
|
||||
$(CPP) -std=c++11 $(COMMON) $(CFLAGS) $^ -o $@ $(LDFLAGS)
|
||||
|
||||
$(OBJDIR)%.o: %.c $(DEPS)
|
||||
$(CC) $(COMMON) $(CFLAGS) -c $< -o $@
|
||||
|
||||
$(OBJDIR)%.o: %.cpp $(DEPS)
|
||||
$(CPP) -std=c++11 $(COMMON) $(CFLAGS) -c $< -o $@
|
||||
|
||||
$(OBJDIR)%.o: %.cu $(DEPS)
|
||||
$(NVCC) $(ARCH) $(COMMON) --compiler-options "$(CFLAGS)" -c $< -o $@
|
||||
|
||||
$(OBJDIR):
|
||||
mkdir -p $(OBJDIR)
|
||||
backup:
|
||||
mkdir -p backup
|
||||
results:
|
||||
mkdir -p results
|
||||
setchmod:
|
||||
chmod +x *.sh
|
||||
|
||||
.PHONY: clean
|
||||
|
||||
clean:
|
||||
rm -rf $(OBJS) $(EXEC) $(LIBNAMESO) $(APPNAMESO)
|
||||
186
docker/Makefile.gpu
Normal file
186
docker/Makefile.gpu
Normal file
@@ -0,0 +1,186 @@
|
||||
|
||||
GPU=1
|
||||
CUDNN=1
|
||||
CUDNN_HALF=0
|
||||
OPENCV=0
|
||||
AVX=0
|
||||
OPENMP=0
|
||||
LIBSO=1
|
||||
ZED_CAMERA=0 # ZED SDK 3.0 and above
|
||||
ZED_CAMERA_v2_8=0 # ZED SDK 2.X
|
||||
|
||||
# set GPU=1 and CUDNN=1 to speedup on GPU
|
||||
# set CUDNN_HALF=1 to further speedup 3 x times (Mixed-precision on Tensor Cores) GPU: Volta, Xavier, Turing and higher
|
||||
# set AVX=1 and OPENMP=1 to speedup on CPU (if error occurs then set AVX=0)
|
||||
|
||||
USE_CPP=0
|
||||
DEBUG=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_61,code=[sm_61,compute_61]
|
||||
|
||||
OS := $(shell uname)
|
||||
|
||||
# Tesla V100
|
||||
# ARCH= -gencode arch=compute_70,code=[sm_70,compute_70]
|
||||
|
||||
# GeForce RTX 2080 Ti, RTX 2080, RTX 2070, Quadro RTX 8000, Quadro RTX 6000, Quadro RTX 5000, Tesla T4, XNOR Tensor Cores
|
||||
# ARCH= -gencode arch=compute_75,code=[sm_75,compute_75]
|
||||
|
||||
# Jetson XAVIER
|
||||
# ARCH= -gencode arch=compute_72,code=[sm_72,compute_72]
|
||||
|
||||
# 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
|
||||
|
||||
# GP100/Tesla P100 - DGX-1
|
||||
# ARCH= -gencode arch=compute_60,code=sm_60
|
||||
|
||||
# For Jetson TX1, Tegra X1, DRIVE CX, DRIVE PX - uncomment:
|
||||
# ARCH= -gencode arch=compute_53,code=[sm_53,compute_53]
|
||||
|
||||
# For Jetson Tx2 or Drive-PX2 uncomment:
|
||||
# ARCH= -gencode arch=compute_62,code=[sm_62,compute_62]
|
||||
|
||||
|
||||
# VPATH=./src/
|
||||
VPATH=./src/:./examples
|
||||
SLIB=libdarknet.so
|
||||
|
||||
EXEC=darknet
|
||||
OBJDIR=./obj/
|
||||
|
||||
ifeq ($(LIBSO), 1)
|
||||
LIBNAMESO=libdarknet.so
|
||||
APPNAMESO=uselib
|
||||
endif
|
||||
|
||||
ifeq ($(USE_CPP), 1)
|
||||
CC=g++
|
||||
else
|
||||
CC=gcc
|
||||
endif
|
||||
|
||||
CPP=g++ -std=c++11
|
||||
NVCC=nvcc
|
||||
OPTS=-Ofast
|
||||
LDFLAGS= -lm -pthread
|
||||
COMMON= -Iinclude/ -I3rdparty/stb/include
|
||||
CFLAGS=-Wall -Wfatal-errors -Wno-unused-result -Wno-unknown-pragmas -fPIC
|
||||
|
||||
ifeq ($(DEBUG), 1)
|
||||
#OPTS= -O0 -g
|
||||
#OPTS= -Og -g
|
||||
COMMON+= -DDEBUG
|
||||
CFLAGS+= -DDEBUG
|
||||
else
|
||||
ifeq ($(AVX), 1)
|
||||
CFLAGS+= -ffp-contract=fast -mavx -mavx2 -msse3 -msse4.1 -msse4.2 -msse4a
|
||||
endif
|
||||
endif
|
||||
|
||||
CFLAGS+=$(OPTS)
|
||||
|
||||
ifneq (,$(findstring MSYS_NT,$(OS)))
|
||||
LDFLAGS+=-lws2_32
|
||||
endif
|
||||
|
||||
ifeq ($(OPENCV), 1)
|
||||
COMMON+= -DOPENCV
|
||||
CFLAGS+= -DOPENCV
|
||||
LDFLAGS+= `pkg-config --libs opencv4 2> /dev/null || pkg-config --libs opencv`
|
||||
COMMON+= `pkg-config --cflags opencv4 2> /dev/null || pkg-config --cflags opencv`
|
||||
endif
|
||||
|
||||
ifeq ($(OPENMP), 1)
|
||||
CFLAGS+= -fopenmp
|
||||
LDFLAGS+= -lgomp
|
||||
endif
|
||||
|
||||
ifeq ($(GPU), 1)
|
||||
COMMON+= -DGPU -I/usr/local/cuda/include/
|
||||
CFLAGS+= -DGPU
|
||||
ifeq ($(OS),Darwin) #MAC
|
||||
LDFLAGS+= -L/usr/local/cuda/lib -lcuda -lcudart -lcublas -lcurand
|
||||
else
|
||||
LDFLAGS+= -L/usr/local/cuda/lib64 -lcuda -lcudart -lcublas -lcurand
|
||||
endif
|
||||
endif
|
||||
|
||||
ifeq ($(CUDNN), 1)
|
||||
COMMON+= -DCUDNN
|
||||
ifeq ($(OS),Darwin) #MAC
|
||||
CFLAGS+= -DCUDNN -I/usr/local/cuda/include
|
||||
LDFLAGS+= -L/usr/local/cuda/lib -lcudnn
|
||||
else
|
||||
CFLAGS+= -DCUDNN -I/usr/local/cudnn/include
|
||||
LDFLAGS+= -L/usr/local/cudnn/lib64 -lcudnn
|
||||
endif
|
||||
endif
|
||||
|
||||
ifeq ($(CUDNN_HALF), 1)
|
||||
COMMON+= -DCUDNN_HALF
|
||||
CFLAGS+= -DCUDNN_HALF
|
||||
ARCH+= -gencode arch=compute_70,code=[sm_70,compute_70]
|
||||
endif
|
||||
|
||||
ifeq ($(ZED_CAMERA), 1)
|
||||
CFLAGS+= -DZED_STEREO -I/usr/local/zed/include
|
||||
ifeq ($(ZED_CAMERA_v2_8), 1)
|
||||
LDFLAGS+= -L/usr/local/zed/lib -lsl_core -lsl_input -lsl_zed
|
||||
#-lstdc++ -D_GLIBCXX_USE_CXX11_ABI=0
|
||||
else
|
||||
LDFLAGS+= -L/usr/local/zed/lib -lsl_zed
|
||||
#-lstdc++ -D_GLIBCXX_USE_CXX11_ABI=0
|
||||
endif
|
||||
endif
|
||||
|
||||
OBJ=image_opencv.o http_stream.o gemm.o utils.o dark_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 gaussian_yolo_layer.o upsample_layer.o lstm_layer.o conv_lstm_layer.o scale_channels_layer.o sam_layer.o
|
||||
ifeq ($(GPU), 1)
|
||||
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
|
||||
endif
|
||||
|
||||
OBJS = $(addprefix $(OBJDIR), $(OBJ))
|
||||
DEPS = $(wildcard src/*.h) Makefile include/darknet.h
|
||||
|
||||
all: $(OBJDIR) backup results setchmod $(EXEC) $(LIBNAMESO) $(APPNAMESO)
|
||||
|
||||
ifeq ($(LIBSO), 1)
|
||||
CFLAGS+= -fPIC
|
||||
|
||||
$(LIBNAMESO): $(OBJDIR) $(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)
|
||||
|
||||
$(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)
|
||||
endif
|
||||
|
||||
$(EXEC): $(OBJS)
|
||||
$(CPP) -std=c++11 $(COMMON) $(CFLAGS) $^ -o $@ $(LDFLAGS)
|
||||
|
||||
$(OBJDIR)%.o: %.c $(DEPS)
|
||||
$(CC) $(COMMON) $(CFLAGS) -c $< -o $@
|
||||
|
||||
$(OBJDIR)%.o: %.cpp $(DEPS)
|
||||
$(CPP) -std=c++11 $(COMMON) $(CFLAGS) -c $< -o $@
|
||||
|
||||
$(OBJDIR)%.o: %.cu $(DEPS)
|
||||
$(NVCC) $(ARCH) $(COMMON) --compiler-options "$(CFLAGS)" -c $< -o $@
|
||||
|
||||
$(OBJDIR):
|
||||
mkdir -p $(OBJDIR)
|
||||
backup:
|
||||
mkdir -p backup
|
||||
results:
|
||||
mkdir -p results
|
||||
setchmod:
|
||||
chmod +x *.sh
|
||||
|
||||
.PHONY: clean
|
||||
|
||||
clean:
|
||||
rm -rf $(OBJS) $(EXEC) $(LIBNAMESO) $(APPNAMESO)
|
||||
32
docker/docker-compose.yml
Normal file
32
docker/docker-compose.yml
Normal file
@@ -0,0 +1,32 @@
|
||||
---
|
||||
version: '3.7'
|
||||
services:
|
||||
|
||||
sidekiq: &darknet_base
|
||||
container_name: ${NAMESPACE}-sidekiq
|
||||
build:
|
||||
context: .
|
||||
dockerfile: Dockerfile
|
||||
image: go-darknet:latest
|
||||
working_dir: /darknet
|
||||
volumes:
|
||||
- darknet-data:/darknet/models
|
||||
command: /darknet/download_data.sh
|
||||
|
||||
darknet:
|
||||
<<: *darknet_base
|
||||
container_name: ${NAMESPACE}-api
|
||||
ports:
|
||||
- "9003:9003"
|
||||
restart: unless-stopped
|
||||
depends_on:
|
||||
- sidekiq
|
||||
command: ["/bin/bash"]
|
||||
# command: ["darknet-server"]
|
||||
|
||||
volumes:
|
||||
darknet-data:
|
||||
driver_opts:
|
||||
type: none
|
||||
o: bind
|
||||
device: ${PWD}/models
|
||||
10
docker/download_data.sh
Executable file
10
docker/download_data.sh
Executable file
@@ -0,0 +1,10 @@
|
||||
#!/bin/sh
|
||||
|
||||
# set -x
|
||||
# set -e
|
||||
|
||||
wget -nc --output-document=sample.jpg https://cdn-images-1.medium.com/max/800/1*EYFejGUjvjPcc4PZTwoufw.jpeg
|
||||
wget -nc --output-document=./models/coco.names https://raw.githubusercontent.com/AlexeyAB/darknet/master/data/coco.names
|
||||
wget -nc --output-document=./models/yolov3.cfg https://raw.githubusercontent.com/AlexeyAB/darknet/master/cfg/yolov3.cfg
|
||||
sed -i -e "\$anames = coco.names" ./models/yolov3.cfg
|
||||
wget -nc --output-document=./models/yolov3.weights https://pjreddie.com/media/files/yolov3.weights
|
||||
2
docker/models/.gitignore
vendored
Normal file
2
docker/models/.gitignore
vendored
Normal file
@@ -0,0 +1,2 @@
|
||||
*
|
||||
!.gitignore
|
||||
Reference in New Issue
Block a user