Simplified PyTorch implementation of one-stage detector: YOLO (You Only Look Once), support multi-gpu training and validating, automatic mixed precision training etc.
torch == 1.8.1
torchmetrics
albumentations
loguru
tqdm
-
Notifications
You must be signed in to change notification settings - Fork 1
Simplified PyTorch implementation of YOLO, support multi-gpu training and validating, automatic mixed precision training etc.
License
zh320/simple-yolo-pytorch
About
Simplified PyTorch implementation of YOLO, support multi-gpu training and validating, automatic mixed precision training etc.
Topics
Resources
License
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published