Skip to content

Commit

Permalink
Implement YOLOv5 model with training pipeline - Add core YOLOv5 archi…
Browse files Browse the repository at this point in the history
…tecture (Conv, Bottleneck, C3, SPPF, Detect) - Implement custom loss function (YOLOLoss) - Add utility functions (bbox_iou, non_max_suppression, xywh2xyxy) - Create basic training loop and data loading structure - Set up main function with argument parsing for easy execution
  • Loading branch information
sanowl committed Jun 21, 2024
1 parent 2211ef5 commit 0efe293
Showing 1 changed file with 383 additions and 443 deletions.
Loading

0 comments on commit 0efe293

Please sign in to comment.