| | |
| | | void train_detection(char *cfgfile, char *weightfile) |
| | | { |
| | | srand(time(0)); |
| | | data_seed = time(0); |
| | | int imgnet = 0; |
| | | char *base = basecfg(cfgfile); |
| | | printf("%s\n", base); |
| | |
| | | plist = get_paths("/home/pjreddie/data/imagenet/det.train.list"); |
| | | }else{ |
| | | plist = get_paths("/home/pjreddie/data/voc/trainall.txt"); |
| | | //plist = get_paths("/home/pjreddie/data/coco/trainval.txt"); |
| | | //plist = get_paths("/home/pjreddie/data/voc/all2007-2012.txt"); |
| | | } |
| | | paths = (char **)list_to_array(plist); |
| | | pthread_t load_thread = load_data_detection_thread(imgs, paths, plist->size, classes, net.w, net.h, side, side, background, &buffer); |
| | |
| | | load_thread = load_data_detection_thread(imgs, paths, plist->size, classes, net.w, net.h, side, side, background, &buffer); |
| | | |
| | | /* |
| | | image im = float_to_image(im_dim, im_dim, 3, train.X.vals[114]); |
| | | draw_detection(im, train.y.vals[114], 7); |
| | | */ |
| | | image im = float_to_image(net.w, net.h, 3, train.X.vals[114]); |
| | | image copy = copy_image(im); |
| | | draw_detection(copy, train.y.vals[114], 7); |
| | | free_image(copy); |
| | | */ |
| | | |
| | | printf("Loaded: %lf seconds\n", sec(clock()-time)); |
| | | time=clock(); |
| | |
| | | fgets(filename, 256, stdin); |
| | | strtok(filename, "\n"); |
| | | image im = load_image_color(filename, im_size, im_size); |
| | | translate_image(im, -128); |
| | | scale_image(im, 1/128.); |
| | | printf("%d %d %d\n", im.h, im.w, im.c); |
| | | float *X = im.data; |
| | | time=clock(); |