README.md

Introduction

Simplified PyTorch implementation of one-stage detector: YOLO (You Only Look Once), support multi-gpu training and validating, automatic mixed precision training etc.

Requirements

torch == 1.8.1
torchmetrics
albumentations
loguru
tqdm