AlexeyAB
2018-08-13 f92b20580a21663c5db9eb8608f8cabd7adbeb10
src/cifar.c
@@ -137,8 +137,8 @@
        axpy_cpu(10, 1, p, 1, pred, 1);
        int index = max_index(pred, 10);
        int class = max_index(test.y.vals[i], 10);
        if(index == class) avg_acc += 1;
        int class_id = max_index(test.y.vals[i], 10);
        if(index == class_id) avg_acc += 1;
        free_image(im);
        printf("%4d: %.2f%%\n", i, 100.*avg_acc/(i+1));
    }
@@ -174,16 +174,16 @@
    data test = load_cifar10_data("data/cifar/cifar-10-batches-bin/test_batch.bin");
    for(i = 0; i < train.X.rows; ++i){
        image im = float_to_image(32, 32, 3, train.X.vals[i]);
        int class = max_index(train.y.vals[i], 10);
        int class_id = max_index(train.y.vals[i], 10);
        char buff[256];
        sprintf(buff, "data/cifar/train/%d_%s",i,labels[class]);
        sprintf(buff, "data/cifar/train/%d_%s",i,labels[class_id]);
        save_image_png(im, buff);
    }
    for(i = 0; i < test.X.rows; ++i){
        image im = float_to_image(32, 32, 3, test.X.vals[i]);
        int class = max_index(test.y.vals[i], 10);
        int class_id = max_index(test.y.vals[i], 10);
        char buff[256];
        sprintf(buff, "data/cifar/test/%d_%s",i,labels[class]);
        sprintf(buff, "data/cifar/test/%d_%s",i,labels[class_id]);
        save_image_png(im, buff);
    }
}