How to create a dataset for YOLO?

nalzcpswc report abuse

Hello! I've been practicing with the YOLO algorithm for some time. I really enjoy it. But so far I have been using only the standard datasets that I found on the Internet. For the image classification problem I could easily make a new dataset from Google pictures or by scraping other sites. But how could I create my own dataset for object detection, when I need annotations in specific format? I want to be able to detect some custom objects too.


Max16Neil report abuse

You should collect images as usual and then add annotations by hand. It’s easy with annotation tools created by the computer vision community. There are several of them, but I personally used LabelImg tool - You need to install it, open your image folder and go through pictures adding rectangles. LabelImg can save annotations in YOLO format directly, or you can use Pascal VOC format.

Add Answer

Need support?

Just drop us an email to ... Show more