kamangir
commited on
Commit
•
a1a8891
1
Parent(s):
60603bc
validating fashion_mnist train - kamangir/bolt#689
Browse files
image_classifier/__init__.py
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
name = "image_classifier"
|
2 |
|
3 |
-
version = "1.1.
|
4 |
|
5 |
description = "fashion-mnist + hugging-face + awesome-bash-cli"
|
|
|
1 |
name = "image_classifier"
|
2 |
|
3 |
+
version = "1.1.65"
|
4 |
|
5 |
description = "fashion-mnist + hugging-face + awesome-bash-cli"
|
image_classifier/classes.py
CHANGED
@@ -155,18 +155,16 @@ class Image_Classifier(object):
|
|
155 |
max_index = test_images.shape[0]
|
156 |
if page_count != -1:
|
157 |
max_index = min(24 * page_count, max_index)
|
158 |
-
|
159 |
-
logger.info(
|
160 |
-
"image_classifier.predict(offset={}): rendering {} frame(s)...".format(
|
161 |
-
offset, max_index
|
162 |
-
)
|
163 |
-
)
|
164 |
for index in tqdm(range(0, max_index, 24)):
|
165 |
self.render(
|
166 |
predictions[index : index + 24],
|
167 |
None if test_labels is None else test_labels[index : index + 24],
|
168 |
test_images[index : index + 24],
|
169 |
-
"{}/
|
|
|
|
|
|
|
170 |
prediction_time,
|
171 |
)
|
172 |
|
|
|
155 |
max_index = test_images.shape[0]
|
156 |
if page_count != -1:
|
157 |
max_index = min(24 * page_count, max_index)
|
158 |
+
logger.info(f"image_classifier.predict(): rendering {max_index} frame(s)...")
|
|
|
|
|
|
|
|
|
|
|
159 |
for index in tqdm(range(0, max_index, 24)):
|
160 |
self.render(
|
161 |
predictions[index : index + 24],
|
162 |
None if test_labels is None else test_labels[index : index + 24],
|
163 |
test_images[index : index + 24],
|
164 |
+
"{}/image_classifier/prediction_output/{:05d}.jpg".format(
|
165 |
+
output_path,
|
166 |
+
int(index / 24),
|
167 |
+
),
|
168 |
prediction_time,
|
169 |
)
|
170 |
|