| | |
|---|
| | | |
|---|
| | | void train_dice(char *cfgfile, char *weightfile) |
|---|
| | | { |
|---|
| | | data_seed = time(0); |
|---|
| | | srand(time(0)); |
|---|
| | | float avg_loss = -1; |
|---|
| | | char *base = basecfg(cfgfile); |
|---|
| | |
|---|
| | | while(1){ |
|---|
| | | ++i; |
|---|
| | | time=clock(); |
|---|
| | | data train = load_data(paths, imgs, plist->size, labels, 6, net.w, net.h); |
|---|
| | | data train = load_data_old(paths, imgs, plist->size, labels, 6, net.w, net.h); |
|---|
| | | printf("Loaded: %lf seconds\n", sec(clock()-time)); |
|---|
| | | |
|---|
| | | time=clock(); |
|---|
| | |
|---|
| | | int m = plist->size; |
|---|
| | | free_list(plist); |
|---|
| | | |
|---|
| | | data val = load_data(paths, m, 0, labels, 6, net.w, net.h); |
|---|
| | | float *acc = network_accuracies(net, val); |
|---|
| | | data val = load_data_old(paths, m, 0, labels, 6, net.w, net.h); |
|---|
| | | float *acc = network_accuracies(net, val, 2); |
|---|
| | | printf("Validation Accuracy: %f, %d images\n", acc[0], m); |
|---|
| | | free_data(val); |
|---|
| | | } |
|---|
| | |
|---|
| | | srand(2222222); |
|---|
| | | int i = 0; |
|---|
| | | char **names = dice_labels; |
|---|
| | | char input[256]; |
|---|
| | | char buff[256]; |
|---|
| | | char *input = buff; |
|---|
| | | int indexes[6]; |
|---|
| | | while(1){ |
|---|
| | | if(filename){ |
|---|
| | |
|---|
| | | }else{ |
|---|
| | | printf("Enter Image Path: "); |
|---|
| | | fflush(stdout); |
|---|
| | | fgets(input, 256, stdin); |
|---|
| | | input = fgets(input, 256, stdin); |
|---|
| | | if(!input) return; |
|---|
| | | strtok(input, "\n"); |
|---|
| | | } |
|---|
| | | image im = load_image_color(input, net.w, net.h); |
|---|
