mirror of
https://github.com/pjreddie/darknet.git
synced 2023-08-10 21:13:14 +03:00
@crossbot I will murder you in your sleep ⚔️
This commit is contained in:
parent
579e588c84
commit
1ef829e585
25
Makefile
25
Makefile
@ -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)
|
||||
|
||||
|
@ -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
|
||||
|
@ -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
|
||||
|
@ -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)
|
@ -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)
|
||||
{
|
@ -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)
|
||||
{
|
@ -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)
|
||||
{
|
@ -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"};
|
||||
|
@ -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;
|
@ -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};
|
||||
|
@ -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"};
|
||||
|
@ -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;
|
@ -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)
|
@ -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
|
||||
|
@ -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)
|
||||
{
|
@ -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;
|
@ -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);
|
@ -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)
|
||||
{
|
@ -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)
|
||||
{
|
@ -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)
|
||||
{
|
@ -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)
|
||||
{
|
@ -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)
|
||||
{
|
@ -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)
|
||||
{
|
@ -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"};
|
||||
|
@ -3,8 +3,8 @@
|
||||
#include "cublas_v2.h"
|
||||
|
||||
extern "C" {
|
||||
#include "activations.h"
|
||||
#include "cuda.h"
|
||||
#include "darknet/activations.h"
|
||||
#include "darknet/cuda.h"
|
||||
}
|
||||
|
||||
|
||||
|
@ -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>
|
||||
|
@ -1,4 +1,4 @@
|
||||
#include "activations.h"
|
||||
#include "darknet/activations.h"
|
||||
|
||||
#include <math.h>
|
||||
#include <stdio.h>
|
||||
|
@ -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)
|
||||
|
@ -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)
|
||||
|
@ -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
|
||||
|
@ -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>
|
||||
|
@ -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)
|
||||
|
@ -1,4 +1,4 @@
|
||||
#include "box.h"
|
||||
#include "darknet/box.h"
|
||||
#include <stdio.h>
|
||||
#include <math.h>
|
||||
#include <stdlib.h>
|
||||
|
@ -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
|
||||
|
@ -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)
|
||||
{
|
||||
|
@ -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>
|
||||
|
@ -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)
|
||||
|
@ -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>
|
||||
|
||||
|
@ -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>
|
||||
|
@ -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>
|
||||
|
@ -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)
|
||||
|
@ -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)
|
||||
|
@ -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>
|
||||
|
||||
|
@ -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>
|
||||
|
@ -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)
|
||||
|
@ -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>
|
||||
|
||||
|
18
src/demo.c
18
src/demo.c
@ -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
|
||||
|
@ -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>
|
||||
|
@ -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>
|
||||
|
||||
|
@ -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)
|
||||
|
@ -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>
|
||||
|
@ -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>
|
||||
|
@ -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)
|
||||
|
@ -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
|
||||
|
12
src/image.c
12
src/image.c
@ -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;
|
||||
|
||||
|
@ -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)
|
||||
|
@ -1,6 +1,6 @@
|
||||
#include <stdlib.h>
|
||||
#include <string.h>
|
||||
#include "list.h"
|
||||
#include "darknet/list.h"
|
||||
|
||||
list *make_list()
|
||||
{
|
||||
|
@ -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>
|
||||
|
||||
|
@ -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>
|
||||
|
@ -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)
|
||||
|
@ -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)
|
||||
|
@ -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)
|
||||
{
|
||||
|
@ -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)
|
||||
|
@ -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)
|
||||
|
@ -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)
|
||||
{
|
||||
|
56
src/parser.c
56
src/parser.c
@ -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;
|
||||
|
@ -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>
|
||||
|
@ -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>
|
||||
|
||||
|
||||
|
@ -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>
|
||||
|
@ -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)
|
||||
|
@ -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>
|
||||
|
||||
|
@ -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>
|
||||
|
@ -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)
|
||||
{
|
||||
|
@ -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)
|
||||
{
|
||||
|
Loading…
Reference in New Issue
Block a user