blame | history | patch | commit | commitdiff | ignore whitespace
Joseph Redmon
2016-11-27 7d34850008b7ed5956bd427e9e74695e1ef46d75 
 src/captcha.c


@@ -28,7 +28,6 @@




void train_captcha(char *cfgfile, char *weightfile)


{


    data_seed = time(0);


    srand(time(0));


    float avg_loss = -1;


    char *base = basecfg(cfgfile);


@@ -38,9 +37,8 @@


        load_weights(&net, weightfile);


    }


    printf("Learning Rate: %g, Momentum: %g, Decay: %g\n", net.learning_rate, net.momentum, net.decay);


    //net.seen=0;


    int imgs = 1024;


    int i = net.seen/imgs;


    int i = *net.seen/imgs;


    int solved = 1;


    list *plist;


    char **labels = get_labels("/data/captcha/reimgs.labels.list");


@@ -85,10 +83,9 @@


        printf("Loaded: %lf seconds\n", sec(clock()-time));


        time=clock();


        float loss = train_network(net, train);


        net.seen += imgs;


        if(avg_loss == -1) avg_loss = loss;


        avg_loss = avg_loss*.9 + loss*.1;


        printf("%d: %f, %f avg, %lf seconds, %d images\n", i, loss, avg_loss, sec(clock()-time), net.seen);


        printf("%d: %f, %f avg, %lf seconds, %d images\n", i, loss, avg_loss, sec(clock()-time), *net.seen);


        free_data(train);


        if(i%100==0){


            char buff[256];


@@ -108,7 +105,8 @@


    srand(2222222);


    int i = 0;


    char **names = get_labels("/data/captcha/reimgs.labels.list");


    char input[256];


    char buff[256];


    char *input = buff;


    int indexes[26];


    while(1){


        if(filename){


@@ -116,7 +114,8 @@


        }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);