From fb9fcfb3ba0057b1196d1b95fd53cdb0b874b1c3 Mon Sep 17 00:00:00 2001
From: AlexeyAB <alexeyab84@gmail.com>
Date: Thu, 17 May 2018 00:04:42 +0000
Subject: [PATCH] Support of .ppm files is realised

---
 src/yolo.c     |    1 +
 src/coco.c     |    1 +
 src/data.c     |    3 +++
 src/detector.c |    3 +++
 4 files changed, 8 insertions(+), 0 deletions(-)

diff --git a/src/coco.c b/src/coco.c
index c95e30d..b8fc346 100644
--- a/src/coco.c
+++ b/src/coco.c
@@ -285,6 +285,7 @@
         find_replace(labelpath, "JPEGImages", "labels", labelpath);
         find_replace(labelpath, ".jpg", ".txt", labelpath);
         find_replace(labelpath, ".JPEG", ".txt", labelpath);
+		find_replace(labelpath, ".ppm", ".txt", labelpath);
 
         int num_labels = 0;
         box_label *truth = read_boxes(labelpath, &num_labels);
diff --git a/src/data.c b/src/data.c
index 4bb1b0f..8da1234 100644
--- a/src/data.c
+++ b/src/data.c
@@ -218,6 +218,7 @@
     find_replace(labelpath, ".jpg", ".txt", labelpath);
     find_replace(labelpath, ".JPG", ".txt", labelpath);
     find_replace(labelpath, ".JPEG", ".txt", labelpath);
+	find_replace(labelpath, ".ppm", ".txt", labelpath);
 
     int count = 0;
     box_label *boxes = read_boxes(labelpath, &count);
@@ -258,6 +259,7 @@
     find_replace(labelpath, ".png", ".txt", labelpath);
     find_replace(labelpath, ".JPG", ".txt", labelpath);
     find_replace(labelpath, ".JPEG", ".txt", labelpath);
+	find_replace(labelpath, ".ppm", ".txt", labelpath);
     int count = 0;
     box_label *boxes = read_boxes(labelpath, &count);
     randomize_boxes(boxes, count);
@@ -309,6 +311,7 @@
 	find_replace(labelpath, ".bmp", ".txt", labelpath);
     find_replace(labelpath, ".JPG", ".txt", labelpath);
     find_replace(labelpath, ".JPEG", ".txt", labelpath);
+	find_replace(labelpath, ".ppm", ".txt", labelpath);
     int count = 0;
 	int i;
     box_label *boxes = read_boxes(labelpath, &count);
diff --git a/src/detector.c b/src/detector.c
index bed7467..5ccb6c5 100644
--- a/src/detector.c
+++ b/src/detector.c
@@ -479,6 +479,7 @@
 		find_replace(labelpath, ".bmp", ".txt", labelpath);
 		find_replace(labelpath, ".JPG", ".txt", labelpath);
 		find_replace(labelpath, ".JPEG", ".txt", labelpath);
+		find_replace(labelpath, ".ppm", ".txt", labelpath);
 
 		int num_labels = 0;
 		box_label *truth = read_boxes(labelpath, &num_labels);
@@ -636,6 +637,7 @@
 			find_replace(labelpath, ".bmp", ".txt", labelpath);
 			find_replace(labelpath, ".JPG", ".txt", labelpath);
 			find_replace(labelpath, ".JPEG", ".txt", labelpath);
+			find_replace(labelpath, ".ppm", ".txt", labelpath);
 			int num_labels = 0;
 			box_label *truth = read_boxes(labelpath, &num_labels);
 			int i, j;
@@ -899,6 +901,7 @@
 		find_replace(labelpath, ".bmp", ".txt", labelpath);
 		find_replace(labelpath, ".JPG", ".txt", labelpath);
 		find_replace(labelpath, ".JPEG", ".txt", labelpath);
+		find_replace(labelpath, ".ppm", ".txt", labelpath);
 		int num_labels = 0;
 		box_label *truth = read_boxes(labelpath, &num_labels);
 		//printf(" new path: %s \n", labelpath);
diff --git a/src/yolo.c b/src/yolo.c
index 238454e..4481165 100644
--- a/src/yolo.c
+++ b/src/yolo.c
@@ -256,6 +256,7 @@
         find_replace(labelpath, "JPEGImages", "labels", labelpath);
         find_replace(labelpath, ".jpg", ".txt", labelpath);
         find_replace(labelpath, ".JPEG", ".txt", labelpath);
+		find_replace(labelpath, ".ppm", ".txt", labelpath);
 
         int num_labels = 0;
         box_label *truth = read_boxes(labelpath, &num_labels);

--
Gitblit v1.10.0