Skip to content

Latest commit

 

History

History
52 lines (35 loc) · 1.03 KB

README.md

File metadata and controls

52 lines (35 loc) · 1.03 KB

Flowers Classifier

Classifies flowers using the Oxford Flowers 102 dataset and MobileNetV2 through TensorFlow.

Developing

First of all, create and activate a venv:

python3 -m venv --system-site-packages ./venv
source ./venv/bin/activate

Then, install the packages:

pip install -r requirements.txt

Once you are done developing, you can exit the virtual env:

deactivate

Usage

Once you're in the virtual env, you can run:

python main.py <image location> <model location>

For example:

python main.py resources/cautleya_spicata.jpg model.h5

That it'll dump the most likely classes of flowers:

cautleya spicata...................99.38%
red ginger.........................0.46%
wallflower.........................0.05%
siam tulip.........................0.03%
monkshood..........................0.02%

You can also specify the following arguments:

  • --top_k to set how many (most likely) classes of flowers you want, defaults to 5