src/detector.c
@@ -99,7 +99,7 @@ args.hue = net.hue; #ifdef OPENCV args.threads = 7; args.threads = 3; IplImage* img = NULL; float max_img_loss = 5; int number_of_lines = 100; @@ -167,7 +167,7 @@ #else loss = train_network(net, train); #endif if (avg_loss < 0) avg_loss = loss; if (avg_loss < 0 || avg_loss != avg_loss) avg_loss = loss; // if(-inf or nan) avg_loss = avg_loss*.9 + loss*.1; i = get_current_batch(net);