Merge pull request #72 from jasrk11/patch-1

Added support for custom data with png image files
This commit is contained in:
Alexey 2017-04-26 16:59:28 +03:00 committed by GitHub
commit da6258bd12
1 changed files with 1 additions and 0 deletions

View File

@ -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);