kamangir
commited on
Commit
•
14f7210
1
Parent(s):
d2b2be7
validating single image predict for fashion_mnist - kamangir/bolt#692
Browse files- abcli/image_classifier.sh +0 -2
- image_classifier/__init__.py +1 -1
- image_classifier/classes.py +1 -1
abcli/image_classifier.sh
CHANGED
|
@@ -106,8 +106,6 @@ function abcli_image_classifier_predict() {
|
|
| 106 |
${@:4}
|
| 107 |
fi
|
| 108 |
|
| 109 |
-
return
|
| 110 |
-
|
| 111 |
if [ "$data_source" == "object" ] ; then
|
| 112 |
cp -v $abcli_object_root/$data_object/*.pyndarray .
|
| 113 |
cp -v $model_path/image_classifier/model/class_names.json .
|
|
|
|
| 106 |
${@:4}
|
| 107 |
fi
|
| 108 |
|
|
|
|
|
|
|
| 109 |
if [ "$data_source" == "object" ] ; then
|
| 110 |
cp -v $abcli_object_root/$data_object/*.pyndarray .
|
| 111 |
cp -v $model_path/image_classifier/model/class_names.json .
|
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.147"
|
| 4 |
|
| 5 |
description = "fashion-mnist + hugging-face + awesome-bash-cli"
|
image_classifier/classes.py
CHANGED
|
@@ -77,7 +77,7 @@ class Image_Classifier(object):
|
|
| 77 |
def predict(
|
| 78 |
self,
|
| 79 |
test_images,
|
| 80 |
-
test_labels,
|
| 81 |
output_path="",
|
| 82 |
page_count=2,
|
| 83 |
):
|
|
|
|
| 77 |
def predict(
|
| 78 |
self,
|
| 79 |
test_images,
|
| 80 |
+
test_labels=None,
|
| 81 |
output_path="",
|
| 82 |
page_count=2,
|
| 83 |
):
|