Linux supported

This commit is contained in:
AlexeyAB 2017-08-03 22:18:00 +03:00
parent 111b749d03
commit 9ff0b7efcb
5 changed files with 27 additions and 3 deletions

View File

@ -2,6 +2,7 @@ GPU=0
CUDNN=0
OPENCV=0
DEBUG=0
OPENMP=0
ARCH= -gencode arch=compute_20,code=[sm_20,sm_21] \
-gencode arch=compute_30,code=sm_30 \
@ -36,6 +37,11 @@ LDFLAGS+= `pkg-config --libs opencv`
COMMON+= `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

6
image_voc.sh Normal file
View File

@ -0,0 +1,6 @@
./darknet detector test ./cfg/voc.data ./cfg/yolo-voc.cfg ./yolo-voc.weights dog.jpg -i 0 -thresh 0.2

6
net_cam_voc.sh Normal file
View File

@ -0,0 +1,6 @@
#rm test_dnn_out.avi
./darknet detector demo ./cfg/voc.data ./cfg/yolo-voc.cfg ./yolo-voc.weights rtsp://admin:admin12345@192.168.0.228:554 -i 0 -thresh 0.24

View File

@ -3,14 +3,14 @@
#include <string.h>
#include <math.h>
#include <assert.h>
#include "unistd.h"
#include <float.h>
#include <limits.h>
#ifdef WIN32
#include "utils.h"
#include "unistd.h"
#else
#include <utils.h>
#include <unistd.h>
#endif
#include "utils.h"
#pragma warning(disable: 4996)

6
video_voc.sh Normal file
View File

@ -0,0 +1,6 @@
./darknet detector demo ./cfg/voc.data ./cfg/yolo-voc.cfg ./yolo-voc.weights test50.mp4 -i 0 -thresh 0.2