From 4403e71b330b42d3cda1e0721fb645cf41bac14f Mon Sep 17 00:00:00 2001
From: AlexeyAB <alexeyab84@gmail.com>
Date: Sun, 20 May 2018 15:43:15 +0000
Subject: [PATCH] Supported files for training: jpg, jpeg, bmp, png, ppm - and UPPER CASE extensions

---
 src/utils.c |   15 +++++++++++++++
 1 files changed, 15 insertions(+), 0 deletions(-)

diff --git a/src/utils.c b/src/utils.c
index 8b1dcf5..d9d796e 100644
--- a/src/utils.c
+++ b/src/utils.c
@@ -183,6 +183,21 @@
     sprintf(output, "%s%s%s", buffer, rep, p+strlen(orig));
 }
 
+void replace_image_to_label(char *input_path, char *output_path) {
+	find_replace(input_path, "images", "labels", output_path);
+	find_replace(output_path, "JPEGImages", "labels", output_path);
+	find_replace(output_path, ".jpg", ".txt", output_path);
+	find_replace(output_path, ".JPG", ".txt", output_path);
+	find_replace(output_path, ".jpeg", ".txt", output_path);
+	find_replace(output_path, ".JPEG", ".txt", output_path);
+	find_replace(output_path, ".png", ".txt", output_path);
+	find_replace(output_path, ".PNG", ".txt", output_path);
+	find_replace(output_path, ".bmp", ".txt", output_path);
+	find_replace(output_path, ".BMP", ".txt", output_path);
+	find_replace(output_path, ".ppm", ".txt", output_path);
+	find_replace(output_path, ".PPM", ".txt", output_path);
+}
+
 float sec(clock_t clocks)
 {
     return (float)clocks/CLOCKS_PER_SEC;

--
Gitblit v1.10.0