| | |
|---|
| | | char *train_images = "/data/voc/train.txt"; |
|---|
| | | char *backup_directory = "/home/pjreddie/backup/"; |
|---|
| | | srand(time(0)); |
|---|
| | | data_seed = time(0); |
|---|
| | | char *base = basecfg(cfgfile); |
|---|
| | | printf("%s\n", base); |
|---|
| | | float avg_loss = -1; |
|---|
| | |
|---|
| | | args.d = &buffer; |
|---|
| | | args.type = REGION_DATA; |
|---|
| | | |
|---|
| | | args.angle = net.angle; |
|---|
| | | args.exposure = net.exposure; |
|---|
| | | args.saturation = net.saturation; |
|---|
| | | args.hue = net.hue; |
|---|
| | | |
|---|
| | | pthread_t load_thread = load_data_in_thread(args); |
|---|
| | | clock_t time; |
|---|
| | | //while(i*imgs < N*120){ |
|---|
| | |
|---|
| | | int nms = 1; |
|---|
| | | float iou_thresh = .5; |
|---|
| | | |
|---|
| | | int nthreads = 2; |
|---|
| | | int nthreads = 8; |
|---|
| | | image *val = calloc(nthreads, sizeof(image)); |
|---|
| | | image *val_resized = calloc(nthreads, sizeof(image)); |
|---|
| | | image *buf = calloc(nthreads, sizeof(image)); |
|---|
| | |
|---|
| | | char buff[256]; |
|---|
| | | char *input = buff; |
|---|
| | | int j; |
|---|
| | | float nms=.5; |
|---|
| | | float nms=.4; |
|---|
| | | box *boxes = calloc(l.side*l.side*l.n, sizeof(box)); |
|---|
| | | float **probs = calloc(l.side*l.side*l.n, sizeof(float *)); |
|---|
| | | for(j = 0; j < l.side*l.side*l.n; ++j) probs[j] = calloc(l.classes, sizeof(float *)); |
|---|
| | |
|---|
| | | save_image(im, "predictions"); |
|---|
| | | show_image(im, "predictions"); |
|---|
| | | |
|---|
| | | show_image(sized, "resized"); |
|---|
| | | free_image(im); |
|---|
| | | free_image(sized); |
|---|
| | | #ifdef OPENCV |
|---|
