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)
|