kamangir commited on
Commit ·
3022254
1
Parent(s): 958fcc3
saving fashion_mnist - kamangir/bolt#692
Browse files- abcli/fashion_mnist.sh +4 -24
- abcli/image_classifier.sh +8 -0
- image_classifier/__init__.py +1 -1
abcli/fashion_mnist.sh
CHANGED
|
@@ -35,30 +35,10 @@ function abcli_fashion_mnist() {
|
|
| 35 |
fi
|
| 36 |
|
| 37 |
if [ "$task" == "predict" ] ; then
|
| 38 |
-
|
| 39 |
-
|
| 40 |
-
|
| 41 |
-
|
| 42 |
-
|
| 43 |
-
if [ "$model_is_object" == 1 ] ; then
|
| 44 |
-
local model_object=$(abcli_clarify_object "$3")
|
| 45 |
-
|
| 46 |
-
abcli_download object $model_object
|
| 47 |
-
|
| 48 |
-
local model_path=$abcli_object_root/$model_object
|
| 49 |
-
else
|
| 50 |
-
local model_name=$(abcli_clarify_arg "$3" fashion-mnist)
|
| 51 |
-
|
| 52 |
-
local model_path=$abcli_path_git/image-classifier/saved_model/$model_name
|
| 53 |
-
fi
|
| 54 |
-
|
| 55 |
-
|
| 56 |
-
# object_1 [name_1] [object]
|
| 57 |
-
|
| 58 |
-
|
| 59 |
-
|
| 60 |
-
|
| 61 |
-
abcli_image_classifier_predict ${@:2}
|
| 62 |
return
|
| 63 |
fi
|
| 64 |
|
|
|
|
| 35 |
fi
|
| 36 |
|
| 37 |
if [ "$task" == "predict" ] ; then
|
| 38 |
+
abcli_huggingface predict \
|
| 39 |
+
image-classifier \
|
| 40 |
+
$(abcli_clarify_arg "$2" fashion-mnist) \
|
| 41 |
+
${@:3}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 42 |
return
|
| 43 |
fi
|
| 44 |
|
abcli/image_classifier.sh
CHANGED
|
@@ -48,6 +48,14 @@ function abcli_image_classifier() {
|
|
| 48 |
return
|
| 49 |
fi
|
| 50 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 51 |
if [ "$task" == "save" ] ; then
|
| 52 |
abcli_huggingface save \
|
| 53 |
image-classifier \
|
|
|
|
| 48 |
return
|
| 49 |
fi
|
| 50 |
|
| 51 |
+
if [ "$task" == "predict" ] ; then
|
| 52 |
+
abcli_huggingface predict \
|
| 53 |
+
image-classifier \
|
| 54 |
+
$(abcli_clarify_arg "$2" image-classifier) \
|
| 55 |
+
${@:3}
|
| 56 |
+
return
|
| 57 |
+
fi
|
| 58 |
+
|
| 59 |
if [ "$task" == "save" ] ; then
|
| 60 |
abcli_huggingface save \
|
| 61 |
image-classifier \
|
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.99"
|
| 4 |
|
| 5 |
description = "fashion-mnist + hugging-face + awesome-bash-cli"
|