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
|