search

ultralytics / yolov3

YOLOv3 in PyTorch > ONNX > CoreML > TFLite
https://docs.ultralytics.com
GNU Affero General Public License v3.0
10.17k stars 3.44k forks source link

YOLOv4 #1115

Closed pedbrgs closed 4 years ago

pedbrgs commented 4 years ago

Hey, Glenn! I saw that the authors of YOLOv4 thanked you for many contributions in it and that you uploaded a yolov4-relu.cfg to this repository. Is it possible to train YOLOv4 through this repository?

Thanks for your attention!

glenn-jocher commented 4 years ago

@pedbrgs yes, you can use yolov4.cfg or yolov4-relu.cfg. The second simply swaps the Mish activations for leaky relu as in yolov3, because Mish is using up great amounts of memory, making it largely impractical to train.

pedbrgs commented 4 years ago

@glenn-jocher Nice! Thank you!

glenn-jocher commented 10 months ago

@pedbrgs you're welcome! It's a team effort - Ultralytics, AlexeyAB, and many others in the YOLO community have made substantial contributions.