Skip to content
Snippets Groups Projects
Commit 3ea51d33 authored by JsPatenaude's avatar JsPatenaude
Browse files

add comments

parent bcacc2d8
No related branches found
No related tags found
No related merge requests found
......@@ -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
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment