kamangir commited on
Commit ·
eb26a46
1
Parent(s): 6e72b38
validating fashion_mnist train - kamangir/bolt#692
Browse files- abcli/image_classifier.sh +1 -1
- image_classifier/__init__.py +1 -1
- image_classifier/classes.py +4 -2
abcli/image_classifier.sh
CHANGED
|
@@ -71,7 +71,7 @@ function abcli_image_classifier_predict() {
|
|
| 71 |
fi
|
| 72 |
|
| 73 |
cp -v ../$data_object/*.pyndarray .
|
| 74 |
-
cp -v ../$model_object/class_names.json .
|
| 75 |
|
| 76 |
python3 -m image_classifier \
|
| 77 |
predict \
|
|
|
|
| 71 |
fi
|
| 72 |
|
| 73 |
cp -v ../$data_object/*.pyndarray .
|
| 74 |
+
cp -v ../$model_object/image_classifier/model/class_names.json .
|
| 75 |
|
| 76 |
python3 -m image_classifier \
|
| 77 |
predict \
|
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.77"
|
| 4 |
|
| 5 |
description = "fashion-mnist + hugging-face + awesome-bash-cli"
|
image_classifier/classes.py
CHANGED
|
@@ -33,7 +33,9 @@ class Image_Classifier(object):
|
|
| 33 |
}
|
| 34 |
|
| 35 |
def load(self, model_path):
|
| 36 |
-
success, self.class_names = file.load_json(
|
|
|
|
|
|
|
| 37 |
if not success:
|
| 38 |
return False
|
| 39 |
|
|
@@ -267,7 +269,7 @@ class Image_Classifier(object):
|
|
| 267 |
self.params["model_size"] = file.size(f"{model_path}/image_classifier/model")
|
| 268 |
|
| 269 |
if not file.save_json(
|
| 270 |
-
f"{model_path}/class_names.json",
|
| 271 |
self.class_names,
|
| 272 |
):
|
| 273 |
return False
|
|
|
|
| 33 |
}
|
| 34 |
|
| 35 |
def load(self, model_path):
|
| 36 |
+
success, self.class_names = file.load_json(
|
| 37 |
+
f"{model_path}/image_classifier/model/class_names.json"
|
| 38 |
+
)
|
| 39 |
if not success:
|
| 40 |
return False
|
| 41 |
|
|
|
|
| 269 |
self.params["model_size"] = file.size(f"{model_path}/image_classifier/model")
|
| 270 |
|
| 271 |
if not file.save_json(
|
| 272 |
+
f"{model_path}/image_classifier/model/class_names.json",
|
| 273 |
self.class_names,
|
| 274 |
):
|
| 275 |
return False
|