From c3b9e4453884f9e44761987fce3b9c6a3dae447b Mon Sep 17 00:00:00 2001 From: Rakesh Jasti Date: Wed, 26 Apr 2017 19:18:19 +0530 Subject: [PATCH] Added support for custom data with png image files --- src/detector.c | 1 + 1 file changed, 1 insertion(+) diff --git a/src/detector.c b/src/detector.c index 5ea6f0c5..2aaa8285 100644 --- a/src/detector.c +++ b/src/detector.c @@ -407,6 +407,7 @@ void validate_detector_recall(char *datacfg, char *cfgfile, char *weightfile) find_replace(labelpath, "JPEGImages", "labels", labelpath); find_replace(labelpath, ".jpg", ".txt", labelpath); find_replace(labelpath, ".JPEG", ".txt", labelpath); + find_replace(labelpath, ".png", ".txt", labelpath); int num_labels = 0; box_label *truth = read_boxes(labelpath, &num_labels);