From 3ea51d33f147f41f6b5b42350e82637816ea8e08 Mon Sep 17 00:00:00 2001
From: JsPatenaude <jean-sebastien.patenaude@polymtl.ca>
Date: Mon, 18 May 2020 13:46:05 -0400
Subject: [PATCH] add comments
---
.../research/dataset/armor_dataset_factory.py | 12 ++++++++++++
1 file changed, 12 insertions(+)
diff --git a/robots-at-robots/research/dataset/armor_dataset_factory.py b/robots-at-robots/research/dataset/armor_dataset_factory.py
index 138de0d..f56736c 100644
--- a/robots-at-robots/research/dataset/armor_dataset_factory.py
+++ b/robots-at-robots/research/dataset/armor_dataset_factory.py
@@ -31,6 +31,18 @@ class ArmorDatasetFactory:
def save_digits_img():
for j, (digit_img, armor, k, p) in enumerate(ArmorDatasetFactory.from_dataset(DJIROCODataset.Final)):
# we ignore the 6s, because the format of their pictures is not accurate and we don't need them
+
+ # usage:
+ # - remove the two if (line 40 and 53-54) to load all images
+ # or
+ # - load bunch of images (not all at the same time, depending if you just want to do some tests)
+ # 1. start with j >= 0 at the top and j >= 500 at the bottom to load the 500 first images
+ # 2. increse these numbers as you load other images (to save time and not load images twice)
+
+
+ # watchout:
+ # some images have chinese symbols in their name, you will either need to change their name of find another workaround
+
if armor.numero != 6 and j >= 5925: # TODO change j value depending where you are in saving (to save time)
# if j >= 50:
# break
--
GitLab