yolo-training-template

A template with tips and scripts for training a YOLO object detection model on a Kaggle dataset and running inference. Useful as a starting point for computer vision projects.

Share on XNo license declared

Overview

This repository is a template for training a YOLO object detection model on a Kaggle dataset and then running inference with the result. It includes a command-line training script, an inference script, a notebook version and a Streamlit web app. Optional preprocessing and NCNN export are available through flags. It is meant as a starting point for computer vision projects, so you can adapt the steps to your own data instead of building the whole pipeline from scratch.

Key features

  • Command-line script to train YOLO on a Kaggle dataset
  • Optional preprocessing and NCNN export flags
  • Separate inference script for trained models
  • Notebook template and Streamlit app for interactive use

Best for

Useful for anyone who wants a working YOLO training setup to adapt to their own computer vision task, including people who prefer a notebook or a web app over the command line.

Upstream
computer-vision-with-marco/yolo-training-template
Fork on GitHub
Guo-astro/yolo-training-template
Upstream stars
511
Category
AI agents and LLM tools
Language
Python
License
No license declaredWithout a license, the author keeps all rights. Ask the upstream owner before reusing the code.
Forked
2025-10-17
Sync status
In syncLast synced 2026-09-29