This toolkit was designed for fast and easy training of YOLO v4 and Tiny YOLO v4 neural networks on the Google Colab GPU. In the beginning you only have to specify the classes from the ImageNetV4 dataset and the samples amount. After that, script will automatically prepare dataset, setting up framework and create most of necessary files.
Toolkit covers the following stages:
- Train_yolov4_imagenet.ipynb:
- Automatic data set formation
- Darknet files preparations and model training
- Predict_yolov4.ipynb:
- Prediction generating
The training set of Imagenet V4 contains 14.6M bounding boxes for 600 object classes on 1.74M images
Dataset description - Overview of Open Images V4
Available image classes - Classes hystoram
Follow this link to start in playground mode Train_yolov4_imagenet.ipynb
Could be useful:
External data: Local Files, Drive, Sheets, and Cloud Storage - How to upload your external data to Google Colab notebook
Connection pushers - How to prevent Google Colab from disconnect during the training.
Darknet - AlexeyAB's version of the Darknet framework
OIDv4_ToolKit - ImagenetV4 dataset downloader
tensorflow-yolov4-tflite - Weights converter