license: cc | |
# Image Classifier | |
An extendable TensorFlow image classifier w/ a Bash cli and Hugging Face integration. | |
--- | |
To see the list of commands added by `image-classifier` complete [installation](#Installation) and type in: | |
``` | |
image_classifier ? | |
fashion_mnist ? | |
``` | |
## Installation | |
`image_classifier` is an [awesome-bash-cli](https://github.com/kamangir/awesome-bash-cli) plugin - to install `image-classifier` first install and configure [awesome-bash-cli](https://github.com/kamangir/awesome-bash-cli), then run: | |
``` | |
abc huggingface clone image-classifier | |
``` | |
To see the list of `image-classifier` models included in the repo, type in | |
``` | |
image_classifier list | |
``` | |
You should see the following items: | |
1. [fashion-mnist](#fashion-mnist) | |
1. intel-image-classifier 🚧 | |
1. vegetable-classifier 🚧 | |
## fashion-mnist | |
![image](./saved_model/fashion-mnist/image_classifier/prediction/00000.jpg) | |
`fashion-mnist` is an `image-classifier` trained on [Fashion-MNIST](https://github.com/zalandoresearch/fashion-mnist). | |
To retrain `fashion-mnist` type in: | |
``` | |
abc select | |
fashion_mnist train | |
abc upload | |
image_classifier list . browser=1,model=object | |
``` | |
You should now see the structure of the network (left) and the [content of the model](https://github.com/kamangir/browser) (right). | |
| ![image](./abcli/assets/fashion_mnist_list.png) | ![image](./abcli/assets/fashion_mnist_browsed.png) | | |
|---|---| | |
You can save this model under a new name by typing in: | |
``` | |
fashion_mnist save new_name . | |
``` | |
/ END |