kamangir commited on
Commit
fa18a17
·
1 Parent(s): 7040b55

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

Browse files
abcli/image_classifier.sh CHANGED
@@ -8,8 +8,8 @@ function abcli_image_classifier() {
8
  local task=$(abcli_unpack_keyword "$1" help)
9
 
10
  if [ "$task" == "help" ] ; then
11
- abcli_help_line "$abcli_cli_name image_classifier describe object_1" \
12
- "describe model object_1."
13
  abcli_help_line "$abcli_cli_name image_classifier install" \
14
  "install image_classifier."
15
  abcli_help_line "$abcli_cli_name image_classifier predict data_1 [name_1] [data=filename/*object/url,model=object/*saved]" \
@@ -31,14 +31,22 @@ function abcli_image_classifier() {
31
  fi
32
 
33
  if [ "$task" == "describe" ] ; then
34
- local model_object=$(abcli_clarify_object "$2" $abcli_object_name)
35
 
36
- abcli_download object $model_object
 
 
 
 
 
 
 
 
37
 
38
  python3 -m image_classifier \
39
  describe \
40
- --model_path $abcli_object_root/$model_object \
41
- ${@:3}
42
 
43
  return
44
  fi
@@ -86,7 +94,7 @@ function abcli_image_classifier_predict() {
86
  local model_path=$(abcli_huggingface get_model_path image-classifier "$model_name" "$options")
87
 
88
  if [ "$model_source" == "object" ] ; then
89
- local model_object=$(python3 -c "print('$model_source'.split('/')[-1])")
90
  abcli_download object $model_object
91
  fi
92
 
 
8
  local task=$(abcli_unpack_keyword "$1" help)
9
 
10
  if [ "$task" == "help" ] ; then
11
+ abcli_help_line "$abcli_cli_name image_classifier describe object_1 [model=object/*saved]" \
12
+ "describe saved/object model object_1."
13
  abcli_help_line "$abcli_cli_name image_classifier install" \
14
  "install image_classifier."
15
  abcli_help_line "$abcli_cli_name image_classifier predict data_1 [name_1] [data=filename/*object/url,model=object/*saved]" \
 
31
  fi
32
 
33
  if [ "$task" == "describe" ] ; then
34
+ local model_name=$2
35
 
36
+ local options=$3
37
+ local model_source=$(abcli_option "$options" "model" saved)
38
+
39
+ local model_path=$(abcli_huggingface get_model_path image-classifier "$model_name" "$options")
40
+
41
+ if [ "$model_source" == "object" ] ; then
42
+ local model_object=$(python3 -c "print('$model_path'.split('/')[-1])")
43
+ abcli_download object $model_object
44
+ fi
45
 
46
  python3 -m image_classifier \
47
  describe \
48
+ --model_path $model_path \
49
+ ${@:4}
50
 
51
  return
52
  fi
 
94
  local model_path=$(abcli_huggingface get_model_path image-classifier "$model_name" "$options")
95
 
96
  if [ "$model_source" == "object" ] ; then
97
+ local model_object=$(python3 -c "print('$model_path'.split('/')[-1])")
98
  abcli_download object $model_object
99
  fi
100
 
image_classifier/__init__.py CHANGED
@@ -1,5 +1,5 @@
1
  name = "image_classifier"
2
 
3
- version = "1.1.164"
4
 
5
  description = "fashion-mnist + hugging-face + awesome-bash-cli"
 
1
  name = "image_classifier"
2
 
3
+ version = "1.1.171"
4
 
5
  description = "fashion-mnist + hugging-face + awesome-bash-cli"