Suyash-Gaurav Logo Image
Suyash Gaurav

Project 1

Conversion of darknet model(.weight) to tensorflow model (.h5)

Project Image

Project Overview

I was learning how to utilize the YOLO technique to train a custom object detection model, which is essentially a darknet model (.weight), but there was an issue because the system was designed for the ESP-32, so they had to use Tensorflow, which used the (.h5)model. Darknet model (.weight) to Tensorflow (.h5) model conversion was something I looked for online but couldn't find, so I created it on my own after reading a good amount of documentation. Future researchers and programmers will undoubtedly benefit from it.

Tools/Package Used

Google Collabotary
Tensorflow documentation
YOLO v4 official paper
Python
Numpy
GIT
Keras