Merge pull request #72 from jasrk11/patch-1

Added support for custom data with png image files
pull/4543/head
Alexey 8 years ago committed by GitHub
commit da6258bd12
  1. 1
      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, "JPEGImages", "labels", labelpath);
find_replace(labelpath, ".jpg", ".txt", labelpath); find_replace(labelpath, ".jpg", ".txt", labelpath);
find_replace(labelpath, ".JPEG", ".txt", labelpath); find_replace(labelpath, ".JPEG", ".txt", labelpath);
find_replace(labelpath, ".png", ".txt", labelpath);
int num_labels = 0; int num_labels = 0;
box_label *truth = read_boxes(labelpath, &num_labels); box_label *truth = read_boxes(labelpath, &num_labels);

Loading…
Cancel
Save