@crossbot I will murder you in your sleep ⚔️

This commit is contained in:
Joseph Redmon 2017-05-31 21:06:35 -07:00
parent 579e588c84
commit 1ef829e585
74 changed files with 404 additions and 393 deletions

View File

@ -12,15 +12,18 @@ ARCH= -gencode arch=compute_20,code=[sm_20,sm_21] \
# 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/
VPATH=./src/:./examples
LIB=libdarknet.a
EXEC=darknet
OBJDIR=./obj/
CC=gcc
NVCC=nvcc
AR=ar
ARFLAGS=-rv
OPTS=-Ofast
LDFLAGS= -lm -pthread
COMMON=
COMMON= -Iinclude/
CFLAGS=-Wall -Wfatal-errors
ifeq ($(DEBUG), 1)
@ -48,19 +51,25 @@ 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 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 segmenter.o regressor.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 lsd.o super.o voxel.o tree.o
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 box.o normalization_layer.o avgpool_layer.o layer.o local_layer.o shortcut_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
EXECOBJA=captcha.o lsd.o super.o voxel.o art.o tag.o cifar.o go.o rnn.o rnn_vid.o compare.o segmenter.o regressor.o classifier.o coco.o dice.o yolo.o detector.o writing.o nightmare.o swag.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 network_kernels.o avgpool_layer_kernels.o
endif
EXECOBJ = $(addprefix $(OBJDIR), $(EXECOBJA))
OBJS = $(addprefix $(OBJDIR), $(OBJ))
DEPS = $(wildcard src/*.h) Makefile
DEPS = $(wildcard include/darknet/*.h) Makefile
all: obj backup results $(EXEC)
all: obj backup results $(LIB) $(EXEC)
$(EXEC): $(OBJS)
$(CC) $(COMMON) $(CFLAGS) $^ -o $@ $(LDFLAGS)
$(EXEC): $(EXECOBJ) $(LIB)
$(CC) $(COMMON) $(CFLAGS) $^ -o $@ $(LDFLAGS) $(LIB)
$(LIB): $(OBJS)
$(AR) $(ARFLAGS) $@ $^
$(OBJDIR)%.o: %.c $(DEPS)
$(CC) $(COMMON) $(CFLAGS) -c $< -o $@
@ -78,5 +87,5 @@ results:
.PHONY: clean
clean:
rm -rf $(OBJS) $(EXEC)
rm -rf $(OBJS) $(LIB) $(EXEC) $(EXECOBJ)

View File

@ -1,6 +1,6 @@
classes=1000
train = /data/imagenet/imagenet1k.train.list
valid = /data/imagenet/imagenet1k.valid.list
valid = /data/imagenet/imagenet1k.train.list
backup = /home/pjreddie/backup/
labels = data/imagenet.labels.list
names = data/imagenet.shortnames.list

View File

@ -5,8 +5,8 @@ subdivisions=1
# Training
# batch=64
# subdivisions=8
width=416
height=416
width=608
height=608
channels=3
momentum=0.9
decay=0.0005

View File

@ -1,9 +1,9 @@
#include "network.h"
#include "utils.h"
#include "parser.h"
#include "option_list.h"
#include "blas.h"
#include "classifier.h"
#include "darknet/network.h"
#include "darknet/utils.h"
#include "darknet/parser.h"
#include "darknet/option_list.h"
#include "darknet/blas.h"
#include "darknet/classifier.h"
#include <sys/time.h>
void demo_art(char *cfgfile, char *weightfile, int cam_index)

View File

@ -1,6 +1,6 @@
#include "network.h"
#include "utils.h"
#include "parser.h"
#include "darknet/network.h"
#include "darknet/utils.h"
#include "darknet/parser.h"
void fix_data_captcha(data d, int mask)
{

View File

@ -1,8 +1,8 @@
#include "network.h"
#include "utils.h"
#include "parser.h"
#include "option_list.h"
#include "blas.h"
#include "darknet/network.h"
#include "darknet/utils.h"
#include "darknet/parser.h"
#include "darknet/option_list.h"
#include "darknet/blas.h"
void train_cifar(char *cfgfile, char *weightfile)
{

View File

@ -1,12 +1,12 @@
#include "network.h"
#include "utils.h"
#include "parser.h"
#include "option_list.h"
#include "blas.h"
#include "assert.h"
#include "classifier.h"
#include "cuda.h"
#include "darknet/network.h"
#include "darknet/utils.h"
#include "darknet/parser.h"
#include "darknet/option_list.h"
#include "darknet/blas.h"
#include "darknet/classifier.h"
#include "darknet/cuda.h"
#include <sys/time.h>
#include <assert.h>
float *get_regression_values(char **labels, int n)
{

View File

@ -1,12 +1,12 @@
#include <stdio.h>
#include "network.h"
#include "detection_layer.h"
#include "cost_layer.h"
#include "utils.h"
#include "parser.h"
#include "box.h"
#include "demo.h"
#include "darknet/network.h"
#include "darknet/detection_layer.h"
#include "darknet/cost_layer.h"
#include "darknet/utils.h"
#include "darknet/parser.h"
#include "darknet/box.h"
#include "darknet/demo.h"
char *coco_classes[] = {"person","bicycle","car","motorcycle","airplane","bus","train","truck","boat","traffic light","fire hydrant","stop sign","parking meter","bench","bird","cat","dog","horse","sheep","cow","elephant","bear","zebra","giraffe","backpack","umbrella","handbag","tie","suitcase","frisbee","skis","snowboard","sports ball","kite","baseball bat","baseball glove","skateboard","surfboard","tennis racket","bottle","wine glass","cup","fork","knife","spoon","bowl","banana","apple","sandwich","orange","broccoli","carrot","hot dog","pizza","donut","cake","chair","couch","potted plant","bed","dining table","toilet","tv","laptop","mouse","remote","keyboard","cell phone","microwave","oven","toaster","sink","refrigerator","book","clock","vase","scissors","teddy bear","hair drier","toothbrush"};

View File

@ -2,11 +2,12 @@
#include <stdlib.h>
#include <stdio.h>
#include "parser.h"
#include "utils.h"
#include "cuda.h"
#include "blas.h"
#include "connected_layer.h"
#include "darknet/parser.h"
#include "darknet/utils.h"
#include "darknet/cuda.h"
#include "darknet/blas.h"
#include "darknet/connected_layer.h"
#include "darknet/convolutional_layer.h"
extern void predict_classifier(char *datacfg, char *cfgfile, char *weightfile, char *filename, int top);
extern void test_detector(char *datacfg, char *cfgfile, char *weightfile, char *filename, float thresh, float hier_thresh, char *outfile, int fullscreen);
@ -172,7 +173,6 @@ void partial(char *cfgfile, char *weightfile, char *outfile, int max)
save_weights_upto(net, outfile, max);
}
#include "convolutional_layer.h"
void rescale_net(char *cfgfile, char *weightfile, char *outfile)
{
gpu_index = -1;

View File

@ -1,12 +1,12 @@
#include "network.h"
#include "region_layer.h"
#include "cost_layer.h"
#include "utils.h"
#include "parser.h"
#include "box.h"
#include "demo.h"
#include "option_list.h"
#include "blas.h"
#include "darknet/network.h"
#include "darknet/region_layer.h"
#include "darknet/cost_layer.h"
#include "darknet/utils.h"
#include "darknet/parser.h"
#include "darknet/box.h"
#include "darknet/demo.h"
#include "darknet/option_list.h"
#include "darknet/blas.h"
static int coco_ids[] = {1,2,3,4,5,6,7,8,9,10,11,13,14,15,16,17,18,19,20,21,22,23,24,25,27,28,31,32,33,34,35,36,37,38,39,40,41,42,43,44,46,47,48,49,50,51,52,53,54,55,56,57,58,59,60,61,62,63,64,65,67,70,72,73,74,75,76,77,78,79,80,81,82,84,85,86,87,88,89,90};

View File

@ -1,6 +1,6 @@
#include "network.h"
#include "utils.h"
#include "parser.h"
#include "darknet/network.h"
#include "darknet/utils.h"
#include "darknet/parser.h"
char *dice_labels[] = {"face1","face2","face3","face4","face5","face6"};

View File

@ -1,9 +1,9 @@
#include "network.h"
#include "utils.h"
#include "parser.h"
#include "option_list.h"
#include "blas.h"
#include "data.h"
#include "darknet/network.h"
#include "darknet/utils.h"
#include "darknet/parser.h"
#include "darknet/option_list.h"
#include "darknet/blas.h"
#include "darknet/data.h"
#include <unistd.h>
int inverted = 1;

View File

@ -1,8 +1,8 @@
#include "network.h"
#include "cost_layer.h"
#include "utils.h"
#include "parser.h"
#include "blas.h"
#include "darknet/network.h"
#include "darknet/cost_layer.h"
#include "darknet/utils.h"
#include "darknet/parser.h"
#include "darknet/blas.h"
/*
void train_lsd3(char *fcfg, char *fweight, char *gcfg, char *gweight, char *acfg, char *aweight, int clear)

View File

@ -1,8 +1,8 @@
#include "network.h"
#include "parser.h"
#include "blas.h"
#include "utils.h"
#include "region_layer.h"
#include "darknet/network.h"
#include "darknet/parser.h"
#include "darknet/blas.h"
#include "darknet/utils.h"
#include "darknet/region_layer.h"
// ./darknet nightmare cfg/extractor.recon.cfg ~/trained/yolo-coco.conv frame6.png -reconstruct -iters 500 -i 3 -lambda .1 -rate .01 -smooth 2

View File

@ -1,11 +1,11 @@
#include "network.h"
#include "utils.h"
#include "parser.h"
#include "option_list.h"
#include "blas.h"
#include "assert.h"
#include "cuda.h"
#include "darknet/network.h"
#include "darknet/utils.h"
#include "darknet/parser.h"
#include "darknet/option_list.h"
#include "darknet/blas.h"
#include "darknet/cuda.h"
#include <sys/time.h>
#include <assert.h>
void train_regressor(char *datacfg, char *cfgfile, char *weightfile, int *gpus, int ngpus, int clear)
{

View File

@ -1,8 +1,8 @@
#include "network.h"
#include "cost_layer.h"
#include "utils.h"
#include "blas.h"
#include "parser.h"
#include "darknet/network.h"
#include "darknet/cost_layer.h"
#include "darknet/utils.h"
#include "darknet/blas.h"
#include "darknet/parser.h"
typedef struct {
float *x;
@ -152,7 +152,7 @@ void train_char_rnn(char *cfgfile, char *weightfile, char *filename, int clear,
}
int inputs = net.inputs;
fprintf(stderr, "Learning Rate: %g, Momentum: %g, Decay: %g\n", net.learning_rate, net.momentum, net.decay);
fprintf(stderr, "Learning Rate: %g, Momentum: %g, Decay: %g, Inputs: %d\n", net.learning_rate, net.momentum, net.decay, inputs);
int batch = net.batch;
int steps = net.time_steps;
if(clear) *net.seen = 0;

View File

@ -1,8 +1,8 @@
#include "network.h"
#include "cost_layer.h"
#include "utils.h"
#include "parser.h"
#include "blas.h"
#include "darknet/network.h"
#include "darknet/cost_layer.h"
#include "darknet/utils.h"
#include "darknet/parser.h"
#include "darknet/blas.h"
#ifdef OPENCV
image get_image_from_stream(CvCapture *cap);

View File

@ -1,11 +1,11 @@
#include "network.h"
#include "utils.h"
#include "parser.h"
#include "option_list.h"
#include "blas.h"
#include "assert.h"
#include "cuda.h"
#include "darknet/network.h"
#include "darknet/utils.h"
#include "darknet/parser.h"
#include "darknet/option_list.h"
#include "darknet/blas.h"
#include "darknet/cuda.h"
#include <sys/time.h>
#include <assert.h>
void train_segmenter(char *datacfg, char *cfgfile, char *weightfile, int *gpus, int ngpus, int clear)
{

View File

@ -1,7 +1,7 @@
#include "network.h"
#include "cost_layer.h"
#include "utils.h"
#include "parser.h"
#include "darknet/network.h"
#include "darknet/cost_layer.h"
#include "darknet/utils.h"
#include "darknet/parser.h"
void train_super(char *cfgfile, char *weightfile, int clear)
{

View File

@ -1,9 +1,9 @@
#include "network.h"
#include "detection_layer.h"
#include "cost_layer.h"
#include "utils.h"
#include "parser.h"
#include "box.h"
#include "darknet/network.h"
#include "darknet/detection_layer.h"
#include "darknet/cost_layer.h"
#include "darknet/utils.h"
#include "darknet/parser.h"
#include "darknet/box.h"
void train_swag(char *cfgfile, char *weightfile)
{

View File

@ -1,6 +1,6 @@
#include "network.h"
#include "utils.h"
#include "parser.h"
#include "darknet/network.h"
#include "darknet/utils.h"
#include "darknet/parser.h"
void train_tag(char *cfgfile, char *weightfile, int clear)
{

View File

@ -1,7 +1,7 @@
#include "network.h"
#include "cost_layer.h"
#include "utils.h"
#include "parser.h"
#include "darknet/network.h"
#include "darknet/cost_layer.h"
#include "darknet/utils.h"
#include "darknet/parser.h"
void extract_voxel(char *lfile, char *rfile, char *prefix)
{

View File

@ -1,6 +1,6 @@
#include "network.h"
#include "utils.h"
#include "parser.h"
#include "darknet/network.h"
#include "darknet/utils.h"
#include "darknet/parser.h"
void train_writing(char *cfgfile, char *weightfile)
{

View File

@ -1,10 +1,10 @@
#include "network.h"
#include "detection_layer.h"
#include "cost_layer.h"
#include "utils.h"
#include "parser.h"
#include "box.h"
#include "demo.h"
#include "darknet/network.h"
#include "darknet/detection_layer.h"
#include "darknet/cost_layer.h"
#include "darknet/utils.h"
#include "darknet/parser.h"
#include "darknet/box.h"
#include "darknet/demo.h"
char *voc_names[] = {"aeroplane", "bicycle", "bird", "boat", "bottle", "bus", "car", "cat", "chair", "cow", "diningtable", "dog", "horse", "motorbike", "person", "pottedplant", "sheep", "sofa", "train", "tvmonitor"};

View File

@ -3,8 +3,8 @@
#include "cublas_v2.h"
extern "C" {
#include "activations.h"
#include "cuda.h"
#include "darknet/activations.h"
#include "darknet/cuda.h"
}

View File

@ -1,8 +1,8 @@
#include "activation_layer.h"
#include "utils.h"
#include "cuda.h"
#include "blas.h"
#include "gemm.h"
#include "darknet/activation_layer.h"
#include "darknet/utils.h"
#include "darknet/cuda.h"
#include "darknet/blas.h"
#include "darknet/gemm.h"
#include <math.h>
#include <stdio.h>

View File

@ -1,4 +1,4 @@
#include "activations.h"
#include "darknet/activations.h"
#include <math.h>
#include <stdio.h>

View File

@ -1,5 +1,5 @@
#include "avgpool_layer.h"
#include "cuda.h"
#include "darknet/avgpool_layer.h"
#include "darknet/cuda.h"
#include <stdio.h>
avgpool_layer make_avgpool_layer(int batch, int w, int h, int c)

View File

@ -3,8 +3,8 @@
#include "cublas_v2.h"
extern "C" {
#include "avgpool_layer.h"
#include "cuda.h"
#include "darknet/avgpool_layer.h"
#include "darknet/cuda.h"
}
__global__ void forward_avgpool_layer_kernel(int n, int w, int h, int c, float *input, float *output)

View File

@ -1,6 +1,6 @@
#include "convolutional_layer.h"
#include "batchnorm_layer.h"
#include "blas.h"
#include "darknet/convolutional_layer.h"
#include "darknet/batchnorm_layer.h"
#include "darknet/blas.h"
#include <stdio.h>
layer make_batchnorm_layer(int batch, int w, int h, int c)
@ -135,10 +135,6 @@ void resize_batchnorm_layer(layer *layer, int w, int h)
void forward_batchnorm_layer(layer l, network net)
{
if(l.type == BATCHNORM) copy_cpu(l.outputs*l.batch, net.input, 1, l.output, 1);
if(l.type == CONNECTED){
l.out_c = l.outputs;
l.out_h = l.out_w = 1;
}
copy_cpu(l.outputs*l.batch, l.output, 1, l.x, 1);
if(net.train){
mean_cpu(l.output, l.batch, l.out_c, l.out_h*l.out_w, l.mean);
@ -193,10 +189,6 @@ void push_batchnorm_layer(layer l)
void forward_batchnorm_layer_gpu(layer l, network net)
{
if(l.type == BATCHNORM) copy_ongpu(l.outputs*l.batch, net.input_gpu, 1, l.output_gpu, 1);
if(l.type == CONNECTED){
l.out_c = l.outputs;
l.out_h = l.out_w = 1;
}
copy_ongpu(l.outputs*l.batch, l.output_gpu, 1, l.x_gpu, 1);
if (net.train) {
#ifdef CUDNN

View File

@ -1,5 +1,6 @@
#include "blas.h"
#include "math.h"
#include "darknet/blas.h"
#include <math.h>
#include <assert.h>
#include <float.h>
#include <stdio.h>

View File

@ -4,9 +4,9 @@
#include <assert.h>
extern "C" {
#include "blas.h"
#include "cuda.h"
#include "utils.h"
#include "darknet/blas.h"
#include "darknet/cuda.h"
#include "darknet/utils.h"
}
__global__ void scale_bias_kernel(float *output, float *biases, int n, int size)

View File

@ -1,4 +1,4 @@
#include "box.h"
#include "darknet/box.h"
#include <stdio.h>
#include <math.h>
#include <stdlib.h>

View File

@ -3,8 +3,8 @@
#include "cublas_v2.h"
extern "C" {
#include "col2im.h"
#include "cuda.h"
#include "darknet/col2im.h"
#include "darknet/cuda.h"
}
// src: https://github.com/BVLC/caffe/blob/master/src/caffe/util/im2col.cu

View File

@ -1,11 +1,11 @@
#include <stdio.h>
#include "network.h"
#include "detection_layer.h"
#include "cost_layer.h"
#include "utils.h"
#include "parser.h"
#include "box.h"
#include "darknet/network.h"
#include "darknet/detection_layer.h"
#include "darknet/cost_layer.h"
#include "darknet/utils.h"
#include "darknet/parser.h"
#include "darknet/box.h"
void train_compare(char *cfgfile, char *weightfile)
{

View File

@ -1,9 +1,9 @@
#include "connected_layer.h"
#include "batchnorm_layer.h"
#include "utils.h"
#include "cuda.h"
#include "blas.h"
#include "gemm.h"
#include "darknet/connected_layer.h"
#include "darknet/batchnorm_layer.h"
#include "darknet/utils.h"
#include "darknet/cuda.h"
#include "darknet/blas.h"
#include "darknet/gemm.h"
#include <math.h>
#include <stdio.h>

View File

@ -3,14 +3,14 @@
#include "cublas_v2.h"
extern "C" {
#include "convolutional_layer.h"
#include "batchnorm_layer.h"
#include "gemm.h"
#include "blas.h"
#include "im2col.h"
#include "col2im.h"
#include "utils.h"
#include "cuda.h"
#include "darknet/convolutional_layer.h"
#include "darknet/batchnorm_layer.h"
#include "darknet/gemm.h"
#include "darknet/blas.h"
#include "darknet/im2col.h"
#include "darknet/col2im.h"
#include "darknet/utils.h"
#include "darknet/cuda.h"
}
__global__ void binarize_kernel(float *x, int n, float *binary)

View File

@ -1,10 +1,10 @@
#include "convolutional_layer.h"
#include "utils.h"
#include "batchnorm_layer.h"
#include "im2col.h"
#include "col2im.h"
#include "blas.h"
#include "gemm.h"
#include "darknet/convolutional_layer.h"
#include "darknet/utils.h"
#include "darknet/batchnorm_layer.h"
#include "darknet/im2col.h"
#include "darknet/col2im.h"
#include "darknet/blas.h"
#include "darknet/gemm.h"
#include <stdio.h>
#include <time.h>

View File

@ -1,7 +1,7 @@
#include "cost_layer.h"
#include "utils.h"
#include "cuda.h"
#include "blas.h"
#include "darknet/cost_layer.h"
#include "darknet/utils.h"
#include "darknet/cuda.h"
#include "darknet/blas.h"
#include <math.h>
#include <string.h>
#include <stdlib.h>

View File

@ -1,9 +1,9 @@
#include "crnn_layer.h"
#include "convolutional_layer.h"
#include "utils.h"
#include "cuda.h"
#include "blas.h"
#include "gemm.h"
#include "darknet/crnn_layer.h"
#include "darknet/convolutional_layer.h"
#include "darknet/utils.h"
#include "darknet/cuda.h"
#include "darknet/blas.h"
#include "darknet/gemm.h"
#include <math.h>
#include <stdio.h>

View File

@ -1,5 +1,5 @@
#include "crop_layer.h"
#include "cuda.h"
#include "darknet/crop_layer.h"
#include "darknet/cuda.h"
#include <stdio.h>
image get_crop_image(crop_layer l)

View File

@ -3,10 +3,10 @@
#include "cublas_v2.h"
extern "C" {
#include "crop_layer.h"
#include "utils.h"
#include "cuda.h"
#include "image.h"
#include "darknet/crop_layer.h"
#include "darknet/utils.h"
#include "darknet/cuda.h"
#include "darknet/image.h"
}
__device__ float get_pixel_kernel(float *image, int w, int h, int x, int y, int c)

View File

@ -2,10 +2,10 @@ int gpu_index = 0;
#ifdef GPU
#include "cuda.h"
#include "utils.h"
#include "blas.h"
#include "assert.h"
#include "darknet/cuda.h"
#include "darknet/utils.h"
#include "darknet/blas.h"
#include <assert.h>
#include <stdlib.h>
#include <time.h>

View File

@ -1,7 +1,7 @@
#include "data.h"
#include "utils.h"
#include "image.h"
#include "cuda.h"
#include "darknet/data.h"
#include "darknet/utils.h"
#include "darknet/image.h"
#include "darknet/cuda.h"
#include <stdio.h>
#include <stdlib.h>

View File

@ -3,15 +3,15 @@
#include "cublas_v2.h"
extern "C" {
#include "convolutional_layer.h"
#include "deconvolutional_layer.h"
#include "batchnorm_layer.h"
#include "gemm.h"
#include "blas.h"
#include "im2col.h"
#include "col2im.h"
#include "utils.h"
#include "cuda.h"
#include "darknet/convolutional_layer.h"
#include "darknet/deconvolutional_layer.h"
#include "darknet/batchnorm_layer.h"
#include "darknet/gemm.h"
#include "darknet/blas.h"
#include "darknet/im2col.h"
#include "darknet/col2im.h"
#include "darknet/utils.h"
#include "darknet/cuda.h"
}
extern "C" void forward_deconvolutional_layer_gpu(layer l, network net)

View File

@ -1,11 +1,12 @@
#include "deconvolutional_layer.h"
#include "convolutional_layer.h"
#include "batchnorm_layer.h"
#include "utils.h"
#include "im2col.h"
#include "col2im.h"
#include "blas.h"
#include "gemm.h"
#include "darknet/deconvolutional_layer.h"
#include "darknet/convolutional_layer.h"
#include "darknet/batchnorm_layer.h"
#include "darknet/utils.h"
#include "darknet/im2col.h"
#include "darknet/col2im.h"
#include "darknet/blas.h"
#include "darknet/gemm.h"
#include <stdio.h>
#include <time.h>

View File

@ -1,12 +1,12 @@
#include "network.h"
#include "detection_layer.h"
#include "region_layer.h"
#include "cost_layer.h"
#include "utils.h"
#include "parser.h"
#include "box.h"
#include "image.h"
#include "demo.h"
#include "darknet/network.h"
#include "darknet/detection_layer.h"
#include "darknet/region_layer.h"
#include "darknet/cost_layer.h"
#include "darknet/utils.h"
#include "darknet/parser.h"
#include "darknet/box.h"
#include "darknet/image.h"
#include "darknet/demo.h"
#include <sys/time.h>
#define DEMO 1

View File

@ -1,10 +1,11 @@
#include "detection_layer.h"
#include "activations.h"
#include "softmax_layer.h"
#include "blas.h"
#include "box.h"
#include "cuda.h"
#include "utils.h"
#include "darknet/detection_layer.h"
#include "darknet/activations.h"
#include "darknet/softmax_layer.h"
#include "darknet/blas.h"
#include "darknet/box.h"
#include "darknet/cuda.h"
#include "darknet/utils.h"
#include <stdio.h>
#include <assert.h>
#include <string.h>

View File

@ -1,6 +1,6 @@
#include "dropout_layer.h"
#include "utils.h"
#include "cuda.h"
#include "darknet/dropout_layer.h"
#include "darknet/utils.h"
#include "darknet/cuda.h"
#include <stdlib.h>
#include <stdio.h>

View File

@ -3,9 +3,9 @@
#include "cublas_v2.h"
extern "C" {
#include "dropout_layer.h"
#include "cuda.h"
#include "utils.h"
#include "darknet/dropout_layer.h"
#include "darknet/cuda.h"
#include "darknet/utils.h"
}
__global__ void yoloswag420blazeit360noscope(float *input, int size, float *rand, float prob, float scale)

View File

@ -1,6 +1,6 @@
#include "gemm.h"
#include "utils.h"
#include "cuda.h"
#include "darknet/gemm.h"
#include "darknet/utils.h"
#include "darknet/cuda.h"
#include <stdlib.h>
#include <stdio.h>
#include <math.h>

View File

@ -1,9 +1,9 @@
#include "gru_layer.h"
#include "connected_layer.h"
#include "utils.h"
#include "cuda.h"
#include "blas.h"
#include "gemm.h"
#include "darknet/gru_layer.h"
#include "darknet/connected_layer.h"
#include "darknet/utils.h"
#include "darknet/cuda.h"
#include "darknet/blas.h"
#include "darknet/gemm.h"
#include <math.h>
#include <stdio.h>

View File

@ -1,4 +1,4 @@
#include "im2col.h"
#include "darknet/im2col.h"
#include <stdio.h>
float im2col_get_pixel(float *im, int height, int width, int channels,
int row, int col, int channel, int pad)

View File

@ -3,8 +3,8 @@
#include "cublas_v2.h"
extern "C" {
#include "im2col.h"
#include "cuda.h"
#include "darknet/im2col.h"
#include "darknet/cuda.h"
}
// src: https://github.com/BVLC/caffe/blob/master/src/caffe/util/im2col.cu

View File

@ -1,14 +1,14 @@
#include "image.h"
#include "utils.h"
#include "blas.h"
#include "cuda.h"
#include "darknet/image.h"
#include "darknet/utils.h"
#include "darknet/blas.h"
#include "darknet/cuda.h"
#include <stdio.h>
#include <math.h>
#define STB_IMAGE_IMPLEMENTATION
#include "stb_image.h"
#include "darknet/stb_image.h"
#define STB_IMAGE_WRITE_IMPLEMENTATION
#include "stb_image_write.h"
#include "darknet/stb_image_write.h"
int windows = 0;

View File

@ -1,5 +1,6 @@
#include "layer.h"
#include "cuda.h"
#include "darknet/layer.h"
#include "darknet/cuda.h"
#include <stdlib.h>
void free_layer(layer l)

View File

@ -1,6 +1,6 @@
#include <stdlib.h>
#include <string.h>
#include "list.h"
#include "darknet/list.h"
list *make_list()
{

View File

@ -1,9 +1,9 @@
#include "local_layer.h"
#include "utils.h"
#include "im2col.h"
#include "col2im.h"
#include "blas.h"
#include "gemm.h"
#include "darknet/local_layer.h"
#include "darknet/utils.h"
#include "darknet/im2col.h"
#include "darknet/col2im.h"
#include "darknet/blas.h"
#include "darknet/gemm.h"
#include <stdio.h>
#include <time.h>

View File

@ -1,6 +1,6 @@
#include "matrix.h"
#include "utils.h"
#include "blas.h"
#include "darknet/matrix.h"
#include "darknet/utils.h"
#include "darknet/blas.h"
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

View File

@ -1,5 +1,5 @@
#include "maxpool_layer.h"
#include "cuda.h"
#include "darknet/maxpool_layer.h"
#include "darknet/cuda.h"
#include <stdio.h>
image get_maxpool_image(maxpool_layer l)

View File

@ -3,8 +3,8 @@
#include "cublas_v2.h"
extern "C" {
#include "maxpool_layer.h"
#include "cuda.h"
#include "darknet/maxpool_layer.h"
#include "darknet/cuda.h"
}
__global__ void forward_maxpool_layer_kernel(int n, int in_h, int in_w, int in_c, int stride, int size, int pad, float *input, float *output, int *indexes)

View File

@ -1,34 +1,34 @@
#include <stdio.h>
#include <time.h>
#include <assert.h>
#include "network.h"
#include "image.h"
#include "data.h"
#include "utils.h"
#include "blas.h"
#include "darknet/network.h"
#include "darknet/image.h"
#include "darknet/data.h"
#include "darknet/utils.h"
#include "darknet/blas.h"
#include "crop_layer.h"
#include "connected_layer.h"
#include "gru_layer.h"
#include "rnn_layer.h"
#include "crnn_layer.h"
#include "local_layer.h"
#include "convolutional_layer.h"
#include "activation_layer.h"
#include "detection_layer.h"
#include "region_layer.h"
#include "normalization_layer.h"
#include "batchnorm_layer.h"
#include "maxpool_layer.h"
#include "reorg_layer.h"
#include "avgpool_layer.h"
#include "cost_layer.h"
#include "softmax_layer.h"
#include "dropout_layer.h"
#include "route_layer.h"
#include "shortcut_layer.h"
#include "parser.h"
#include "data.h"
#include "darknet/crop_layer.h"
#include "darknet/connected_layer.h"
#include "darknet/gru_layer.h"
#include "darknet/rnn_layer.h"
#include "darknet/crnn_layer.h"
#include "darknet/local_layer.h"
#include "darknet/convolutional_layer.h"
#include "darknet/activation_layer.h"
#include "darknet/detection_layer.h"
#include "darknet/region_layer.h"
#include "darknet/normalization_layer.h"
#include "darknet/batchnorm_layer.h"
#include "darknet/maxpool_layer.h"
#include "darknet/reorg_layer.h"
#include "darknet/avgpool_layer.h"
#include "darknet/cost_layer.h"
#include "darknet/softmax_layer.h"
#include "darknet/dropout_layer.h"
#include "darknet/route_layer.h"
#include "darknet/shortcut_layer.h"
#include "darknet/parser.h"
#include "darknet/data.h"
load_args get_base_args(network net)
{

View File

@ -7,32 +7,32 @@ extern "C" {
#include <time.h>
#include <assert.h>
#include "network.h"
#include "data.h"
#include "utils.h"
#include "parser.h"
#include "darknet/network.h"
#include "darknet/data.h"
#include "darknet/utils.h"
#include "darknet/parser.h"
#include "crop_layer.h"
#include "connected_layer.h"
#include "rnn_layer.h"
#include "gru_layer.h"
#include "crnn_layer.h"
#include "detection_layer.h"
#include "region_layer.h"
#include "convolutional_layer.h"
#include "activation_layer.h"
#include "maxpool_layer.h"
#include "reorg_layer.h"
#include "avgpool_layer.h"
#include "normalization_layer.h"
#include "batchnorm_layer.h"
#include "cost_layer.h"
#include "local_layer.h"
#include "softmax_layer.h"
#include "dropout_layer.h"
#include "route_layer.h"
#include "shortcut_layer.h"
#include "blas.h"
#include "darknet/crop_layer.h"
#include "darknet/connected_layer.h"
#include "darknet/rnn_layer.h"
#include "darknet/gru_layer.h"
#include "darknet/crnn_layer.h"
#include "darknet/detection_layer.h"
#include "darknet/region_layer.h"
#include "darknet/convolutional_layer.h"
#include "darknet/activation_layer.h"
#include "darknet/maxpool_layer.h"
#include "darknet/reorg_layer.h"
#include "darknet/avgpool_layer.h"
#include "darknet/normalization_layer.h"
#include "darknet/batchnorm_layer.h"
#include "darknet/cost_layer.h"
#include "darknet/local_layer.h"
#include "darknet/softmax_layer.h"
#include "darknet/dropout_layer.h"
#include "darknet/route_layer.h"
#include "darknet/shortcut_layer.h"
#include "darknet/blas.h"
}
void forward_network_gpu(network net)

View File

@ -1,5 +1,6 @@
#include "normalization_layer.h"
#include "blas.h"
#include "darknet/normalization_layer.h"
#include "darknet/blas.h"
#include <stdio.h>
layer make_normalization_layer(int batch, int w, int h, int c, int size, float alpha, float beta, float kappa)

View File

@ -1,8 +1,8 @@
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include "option_list.h"
#include "utils.h"
#include "darknet/option_list.h"
#include "darknet/utils.h"
list *read_data_cfg(char *filename)
{

View File

@ -1,35 +1,35 @@
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <assert.h>
#include "activation_layer.h"
#include "activations.h"
#include "assert.h"
#include "avgpool_layer.h"
#include "batchnorm_layer.h"
#include "blas.h"
#include "connected_layer.h"
#include "deconvolutional_layer.h"
#include "convolutional_layer.h"
#include "cost_layer.h"
#include "crnn_layer.h"
#include "crop_layer.h"
#include "detection_layer.h"
#include "dropout_layer.h"
#include "gru_layer.h"
#include "list.h"
#include "local_layer.h"
#include "maxpool_layer.h"
#include "normalization_layer.h"
#include "option_list.h"
#include "parser.h"
#include "region_layer.h"
#include "reorg_layer.h"
#include "rnn_layer.h"
#include "route_layer.h"
#include "shortcut_layer.h"
#include "softmax_layer.h"
#include "utils.h"
#include "darknet/activation_layer.h"
#include "darknet/activations.h"
#include "darknet/avgpool_layer.h"
#include "darknet/batchnorm_layer.h"
#include "darknet/blas.h"
#include "darknet/connected_layer.h"
#include "darknet/deconvolutional_layer.h"
#include "darknet/convolutional_layer.h"
#include "darknet/cost_layer.h"
#include "darknet/crnn_layer.h"
#include "darknet/crop_layer.h"
#include "darknet/detection_layer.h"
#include "darknet/dropout_layer.h"
#include "darknet/gru_layer.h"
#include "darknet/list.h"
#include "darknet/local_layer.h"
#include "darknet/maxpool_layer.h"
#include "darknet/normalization_layer.h"
#include "darknet/option_list.h"
#include "darknet/parser.h"
#include "darknet/region_layer.h"
#include "darknet/reorg_layer.h"
#include "darknet/rnn_layer.h"
#include "darknet/route_layer.h"
#include "darknet/shortcut_layer.h"
#include "darknet/softmax_layer.h"
#include "darknet/utils.h"
typedef struct{
char *type;

View File

@ -1,9 +1,10 @@
#include "region_layer.h"
#include "activations.h"
#include "blas.h"
#include "box.h"
#include "cuda.h"
#include "utils.h"
#include "darknet/region_layer.h"
#include "darknet/activations.h"
#include "darknet/blas.h"
#include "darknet/box.h"
#include "darknet/cuda.h"
#include "darknet/utils.h"
#include <stdio.h>
#include <assert.h>
#include <string.h>

View File

@ -1,6 +1,7 @@
#include "reorg_layer.h"
#include "cuda.h"
#include "blas.h"
#include "darknet/reorg_layer.h"
#include "darknet/cuda.h"
#include "darknet/blas.h"
#include <stdio.h>

View File

@ -1,9 +1,9 @@
#include "rnn_layer.h"
#include "connected_layer.h"
#include "utils.h"
#include "cuda.h"
#include "blas.h"
#include "gemm.h"
#include "darknet/rnn_layer.h"
#include "darknet/connected_layer.h"
#include "darknet/utils.h"
#include "darknet/cuda.h"
#include "darknet/blas.h"
#include "darknet/gemm.h"
#include <math.h>
#include <stdio.h>

View File

@ -1,6 +1,7 @@
#include "route_layer.h"
#include "cuda.h"
#include "blas.h"
#include "darknet/route_layer.h"
#include "darknet/cuda.h"
#include "darknet/blas.h"
#include <stdio.h>
route_layer make_route_layer(int batch, int n, int *input_layers, int *input_sizes)

View File

@ -1,6 +1,7 @@
#include "shortcut_layer.h"
#include "cuda.h"
#include "blas.h"
#include "darknet/shortcut_layer.h"
#include "darknet/cuda.h"
#include "darknet/blas.h"
#include <stdio.h>
#include <assert.h>

View File

@ -1,6 +1,7 @@
#include "softmax_layer.h"
#include "blas.h"
#include "cuda.h"
#include "darknet/softmax_layer.h"
#include "darknet/blas.h"
#include "darknet/cuda.h"
#include <float.h>
#include <math.h>
#include <stdlib.h>

View File

@ -1,8 +1,8 @@
#include <stdio.h>
#include <stdlib.h>
#include "tree.h"
#include "utils.h"
#include "data.h"
#include "darknet/tree.h"
#include "darknet/utils.h"
#include "darknet/data.h"
void change_leaves(tree *t, char *leaf_list)
{

View File

@ -7,7 +7,7 @@
#include <float.h>
#include <limits.h>
#include "utils.h"
#include "darknet/utils.h"
int *read_intlist(char *gpu_list, int *ngpus, int d)
{