kamangir commited on
Commit
21ce60a
1 Parent(s): fa18a17

validating single image predict for fashion_mnist - kamangir/bolt#692

Browse files
image_classifier/__init__.py CHANGED
@@ -1,5 +1,5 @@
1
  name = "image_classifier"
2
 
3
- version = "1.1.171"
4
 
5
  description = "fashion-mnist + hugging-face + awesome-bash-cli"
 
1
  name = "image_classifier"
2
 
3
+ version = "1.1.172"
4
 
5
  description = "fashion-mnist + hugging-face + awesome-bash-cli"
image_classifier/classes.py CHANGED
@@ -54,17 +54,15 @@ class Image_Classifier(object):
54
  f"{model_path}/image_classifier/model"
55
  )
56
  except:
57
- from abcli.logging import crash_report
58
-
59
- crash_report("image_classifier.load({}) failed".format(model_path))
60
  return False
61
 
62
  logger.info(
63
- "{}.load({}x{}:{}): {}{} class(es): {}".format(
64
  self.__class__.__name__,
 
65
  self.params["window_size"],
66
  self.params["window_size"],
67
- path.name(model_path),
68
  "convnet - " if self.params["convnet"] else "",
69
  len(self.class_names),
70
  ",".join(self.class_names),
@@ -196,9 +194,7 @@ class Image_Classifier(object):
196
  )
197
  )
198
  except:
199
- from abcli.logging import crash_report
200
-
201
- crash_report("image_classifier.predict_frame() crashed.")
202
  return False, -1
203
 
204
  prediction_time = time.time() - prediction_time
 
54
  f"{model_path}/image_classifier/model"
55
  )
56
  except:
57
+ crash_report("-f image_classifier: load({model_path}): failed.")
 
 
58
  return False
59
 
60
  logger.info(
61
+ "{}.load({}): {}x{} {}{} class(es): {}".format(
62
  self.__class__.__name__,
63
+ model_path,
64
  self.params["window_size"],
65
  self.params["window_size"],
 
66
  "convnet - " if self.params["convnet"] else "",
67
  len(self.class_names),
68
  ",".join(self.class_names),
 
194
  )
195
  )
196
  except:
197
+ crash_report("-image_classifier: predict_frame(): failed.")
 
 
198
  return False, -1
199
 
200
  prediction_time = time.time() - prediction_time