kamangir
commited on
Commit
·
02c423d
1
Parent(s):
08b815a
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.52"
|
4 |
|
5 |
description = "fashion-mnist + hugging-face + awesome-bash-cli"
|
image_classifier/classes.py
CHANGED
@@ -268,7 +268,7 @@ class Image_Classifier(object):
|
|
268 |
"image_classifier",
|
269 |
self.object_name,
|
270 |
string.pretty_bytes(self.model_size) if self.model_size else "",
|
271 |
-
string.
|
272 |
"/".join(string.shorten(self.class_names)),
|
273 |
"took {} / frame".format(
|
274 |
string.pretty_duration(
|
@@ -322,7 +322,7 @@ class Image_Classifier(object):
|
|
322 |
if convnet
|
323 |
else (window_size, window_size)
|
324 |
)
|
325 |
-
logger.info(f"
|
326 |
|
327 |
if convnet and not color:
|
328 |
train_images = np.expand_dims(train_images, axis=3)
|
|
|
268 |
"image_classifier",
|
269 |
self.object_name,
|
270 |
string.pretty_bytes(self.model_size) if self.model_size else "",
|
271 |
+
string.pretty_shape(self.input_shape),
|
272 |
"/".join(string.shorten(self.class_names)),
|
273 |
"took {} / frame".format(
|
274 |
string.pretty_duration(
|
|
|
322 |
if convnet
|
323 |
else (window_size, window_size)
|
324 |
)
|
325 |
+
logger.info(f"input:{string.pretty_shape(input_shape)}")
|
326 |
|
327 |
if convnet and not color:
|
328 |
train_images = np.expand_dims(train_images, axis=3)
|