Skip to content

A clean, modular implementation of the Yolov7 model family, which uses the official pretrained weights, with utilities for training the model on custom (non-COCO) tasks.

License

Notifications You must be signed in to change notification settings

Chris-hughes10/Yolov7-training

Repository files navigation

Yolov7-training

A clean, modular implementation of the Yolov7 model family, which uses the official pretrained weights, with utilities for training the model on custom (non-COCO) tasks.

This Repo includes:

  • PyTorch implementations of the Yolov7 models defined in the original paper
  • An implementation of the Yolov7 loss function
  • Components to create a modular data loading pipeline, such as a PyTorch dataset wrapper and custom Mosaic and Mixup implementations
  • Evaluation utilities using PyCOCOTools
  • A Generic model Trainer, which can be extended or adapted to new tasks

Examples of use are available here, and more detail is provided in this blog post.

Installation

This repo can be installed as a package using the following command:

pip install git+https://github.com/Chris-hughes10/Yolov7-training.git

Usage

Whilst we would recommend using the official implementation if you wish to exactly reproduce the published results on COCO, we find that this implementation is more flexible to apply, and extend, to custom domains.

The aim of this implementation is provide a clean and clear starting point for anyone wishing to experiment with Yolov7 in their own custom training scripts, as well as providing more transparency around the techniques that were used during training in the original implementation.

About

A clean, modular implementation of the Yolov7 model family, which uses the official pretrained weights, with utilities for training the model on custom (non-COCO) tasks.

Resources

License

Stars

Watchers

Forks

Packages

No packages published