src/yolo_layer.c
@@ -206,7 +206,7 @@ getchar(); continue; // if label contains class_id more than number of classes in the cfg-file } if(!truth.x) break; if(!truth.x) break; // continue; float iou = box_iou(pred, truth); if (iou > best_iou) { best_iou = iou; @@ -237,7 +237,7 @@ int class_id = state.truth[t*(4 + 1) + b*l.truths + 4]; if (class_id >= l.classes) continue; // if label contains class_id more than number of classes in the cfg-file if(!truth.x) break; if(!truth.x) break; // continue; float best_iou = 0; int best_n = 0; i = (truth.x * l.w);