This commit is contained in:
Joseph Redmon
2017-06-01 20:31:13 -07:00
parent 1ef829e585
commit 56d69e73ab
87 changed files with 915 additions and 888 deletions

View File

@ -1,9 +1,5 @@
#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.h"
#include <sys/time.h>
void demo_art(char *cfgfile, char *weightfile, int cam_index)

View File

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

View File

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

View File

@ -1,10 +1,5 @@
#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 "darknet.h"
#include <sys/time.h>
#include <assert.h>
@ -37,11 +32,7 @@ void train_classifier(char *datacfg, char *cfgfile, char *weightfile, int *gpus,
#ifdef GPU
cuda_set_device(gpus[i]);
#endif
nets[i] = parse_network_cfg(cfgfile);
if(weightfile){
load_weights(&nets[i], weightfile);
}
if(clear) *nets[i].seen = 0;
nets[i] = load_network(cfgfile, weightfile, clear);
nets[i].learning_rate *= ngpus;
}
srand(time(0));

View File

@ -1,12 +1,6 @@
#include <stdio.h>
#include "darknet.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"
#include <stdio.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

@ -1,14 +1,9 @@
#include "darknet.h"
#include <time.h>
#include <stdlib.h>
#include <stdio.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);
extern void run_voxel(int argc, char **argv);

View File

@ -1,12 +1,4 @@
#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"
#include "darknet.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,4 @@
#include "darknet/network.h"
#include "darknet/utils.h"
#include "darknet/parser.h"
#include "darknet.h"
char *dice_labels[] = {"face1","face2","face3","face4","face5","face6"};

View File

@ -1,9 +1,5 @@
#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 "darknet.h"
#include <unistd.h>
int inverted = 1;

View File

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

View File

@ -1,8 +1,4 @@
#include "darknet/network.h"
#include "darknet/parser.h"
#include "darknet/blas.h"
#include "darknet/utils.h"
#include "darknet/region_layer.h"
#include "darknet.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,9 +1,4 @@
#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 "darknet.h"
#include <sys/time.h>
#include <assert.h>

View File

@ -1,8 +1,4 @@
#include "darknet/network.h"
#include "darknet/cost_layer.h"
#include "darknet/utils.h"
#include "darknet/blas.h"
#include "darknet/parser.h"
#include "darknet.h"
typedef struct {
float *x;

View File

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

View File

@ -1,9 +1,4 @@
#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 "darknet.h"
#include <sys/time.h>
#include <assert.h>
@ -153,7 +148,9 @@ void predict_segmenter(char *datafile, char *cfgfile, char *weightfile, char *fi
image rgb = mask_to_rgb(m);
show_image(sized, "orig");
show_image(rgb, "pred");
#ifdef OPENCV
cvWaitKey(0);
#endif
printf("Predicted: %f\n", predictions[0]);
printf("%s: Predicted in %f seconds.\n", input, sec(clock()-time));
free_image(im);

View File

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

View File

@ -1,9 +1,5 @@
#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.h"
#include <sys/time.h>
void train_swag(char *cfgfile, char *weightfile)
{

View File

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

View File

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

View File

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

View File

@ -1,10 +1,4 @@
#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"
#include "darknet.h"
char *voc_names[] = {"aeroplane", "bicycle", "bird", "boat", "bottle", "bus", "car", "cat", "chair", "cow", "diningtable", "dog", "horse", "motorbike", "person", "pottedplant", "sheep", "sofa", "train", "tvmonitor"};