How to Train YOLOv4 on a Custom Dataset (PyTorch)

This tutorial shows you how to train a YOLOv4 object detection model on your own dataset using free tools (Roboflow, Google Colab). The YOLOv4 implementation is in PyTorch, but the model can be exported with ONNX to TensorFlow. Create a free Roboflow account: Access to the Google Colab notebook: Convert to ONNX and TensorFlow:
Back to Top