Skip to content

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

Notifications You must be signed in to change notification settings

luizguilhermefr/FlowersClassifier

Repository files navigation

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

About

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

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages