Output logfile while training

Author: arnoldkalmbach

examples/detector.c

@@ -1,3 +1,4 @@
+#include <stdio.h>
 #include "darknet.h"
 
 static int coco_ids[] = {1,2,3,4,5,6,7,8,9,10,11,13,14,15,16,17,18,19,20,21,22,23,24,25,27,28,31,32,33,34,35,36,37,38,39,40,41,42,43,44,46,47,48,49,50,51,52,53,54,55,56,57,58,59,60,61,62,63,64,65,67,70,72,73,74,75,76,77,78,79,80,81,82,84,85,86,87,88,89,90};
@@ -58,7 +59,17 @@ void train_detector(char *datacfg, char *cfgfile, char *weightfile, int *gpus, i
     pthread_t load_thread = load_data(args);
     double time;
     int count = 0;
-    //while(i*imgs < N*120){
+    
+    const char* logfname = "/train_log.csv";
+    char* logfpath = malloc(strlen(backup_directory) + strlen(logfname));
+    strcpy(logfpath, backup_directory);
+    strcat(logfpath, logfname);
+    FILE *logfile = fopen(logfpath, "w");
+    printf("Writing log to %s\n", logfpath);
+    if (logfile == NULL)
+        printf("Could not open logfile for writing!!\n");
+    fprintf(logfile, "Batch, Loss, Smooth Loss, Rate, Batch Time, Images Seen\n");
+    fflush(logfile);
     while(get_current_batch(net) < net->max_batches){
         if(l.random && count++%10 == 0){
             printf("Resizing\n");
@@ -127,6 +138,8 @@ void train_detector(char *datacfg, char *cfgfile, char *weightfile, int *gpus, i
 
         i = get_current_batch(net);
         printf("%ld: %f, %f avg, %f rate, %lf seconds, %d images\n", get_current_batch(net), loss, avg_loss, get_current_rate(net), what_time_is_it_now()-time, i*imgs);
+        fprintf(logfile, "%ld, %f, %f, %f, %lf, %d\n", get_current_batch(net), loss, avg_loss, get_current_rate(net), what_time_is_it_now()-time, i*imgs);
+        fflush(logfile);
         if(i%100==0){
 #ifdef GPU
             if(ngpus != 1) sync_nets(nets, ngpus, 0);