Added support for custom data with png image files

This commit is contained in:
Rakesh Jasti
2017-04-26 19:18:19 +05:30
committed by GitHub
parent 7667154f1e
commit c3b9e44538

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