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.51"
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.pretty_size(self.input_shape),
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"input_shape:{string.pretty_size(input_shape)}")
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)