Object detection tools to detect tow trucks

User-Setup

Join the Discord server for help

Dev-Setup

Prerequisites:

• Git - Free

Setup

git clone https://github.com/Haste171/tow-truck-object-detection.git

Install Requirements

poetry install

Activate Environment

poetry shell

Setup ClearML for Logging

clearml-init

Access ClearML

Go to https://app.clear.ml/projects to access the project created once training starts to view logs etc.

Usage

Training

python main.py train --dataset /path/to/dataset --model_size medium --name TowTruckModel --clear_ml_name TowTruckObjectDetection --patience 10

Inference

python main.py infer --image /path/to/image.jpg