/yolo-custom-object-detector

Training Yolo on custom dataset to detect Rubik's Cube

Primary LanguageJava

Building a custom object detector using YOLO

This repository contains the code and the dataset for the tutorials (Part1 and Part2) I wrote about making custome object detector using YOLO in Java and Python. It also contains a dockerfile to build a docker image contains darknet framwork, OpenCV 3.3, and CUDA.