Supported files for training: jpg, jpeg, bmp, png, ppm - and UPPER CASE extensions

This commit is contained in:
AlexeyAB
2018-05-20 18:43:15 +03:00
parent 175c68b9b3
commit 4403e71b33
6 changed files with 25 additions and 79 deletions

View File

@ -281,11 +281,7 @@ void validate_coco_recall(char *cfgfile, char *weightfile)
if (nms) do_nms(boxes, probs, side*side*l.n, 1, nms_thresh);
char labelpath[4096];
find_replace(path, "images", "labels", labelpath);
find_replace(labelpath, "JPEGImages", "labels", labelpath);
find_replace(labelpath, ".jpg", ".txt", labelpath);
find_replace(labelpath, ".JPEG", ".txt", labelpath);
find_replace(labelpath, ".ppm", ".txt", labelpath);
replace_image_to_label(path, labelpath);
int num_labels = 0;
box_label *truth = read_boxes(labelpath, &num_labels);