Skip to content

This repository is a playpen area to work on some machine learning pipeline demos for image classification (tensorflow) and bounding box detection (yolo5)

Notifications You must be signed in to change notification settings

ashtondavi/machine_learning_demos

Repository files navigation

Overview

This repository is a playpen area to work on some machine learning pipeline demos for image classification and bounding box detection.

Usage (from vscode)

Set up a virtual environment by running "python -m venv .env"

Switch the the virtual environment

Install the demos packages from the repos base directory by running "pip install -e ."

copy and paste one of the commands below inot the powershell to run with example data

Commands to run pipelines from windows cmd or vscode

python demos/pipelines/bbox_detection.py --folder "./example_data/images" python demos/pipelines/classification.py --output "./classifier" --epochs "10"

About

This repository is a playpen area to work on some machine learning pipeline demos for image classification (tensorflow) and bounding box detection (yolo5)

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages