/DhakaAI-yolov5

YOLOv5 in PyTorch > ONNX > CoreML > TFLite

Primary LanguagePythonGNU General Public License v3.0GPL-3.0

This repository contains the 9th place (out of 241 teams) solution of the DhakaAI traffic detection competition. We incorporated our evolved hyperparameters by training several configurations of yolov5 models on the competition dataset using the ultralytics YOLOv5 repository. Our approach in the competition is briefly described in this poster. The data preprocessing of yolov5 is same as yolov3 and codes are available here. The training and inference codes are provided here. All code and models are under active development, and are subject to modification.

Inferences