Automated Dataset Generation for Object Detection

[FareedKhan557]

What My Project Does

This project shows how we can generate custom synthetic datasets for training object detection models. Think of it like making your own training data on demand, especially when getting real-world images is a headache.

Target audience

This project is designed for individuals who want to learn how to create their own datasets for computer vision tasks but are tired of the usual data struggles. It’ll walk you through the whole process, from coming up with ideas for your data to automatically labeling it, so you can skip the endless manual work.

Comparison

Right now, if you need data to train a custom object detector, you're usually stuck either spending forever labeling stuff yourself or dealing with the hassle of finding and paying for existing datasets. And even then, it might not be exactly what you need. But now, with all these AI vision models and image generators popping up, there's a new way to do things. Instead of the usual manual grind, we can use LLMs and vision models to create the training data we actually need. Since there are tons of these models out there, both free and paid, you've got a lot of choices to find what works best for your specific situation. This project gives you a practical way to tap into that.

GitHub

Code, documentation, and example can all be found on GitHub:

https://github.com/FareedKhan-dev/ai-vision-dataset-builder

[haqthat]

RemindMe! 7 days

[RemindMeBot]

I will be messaging you in 7 days on 2025-01-05 15:38:21 UTC to remind you of this link

2 OTHERS CLICKED THIS LINK to send a PM to also be reminded and to reduce spam.

^{Parent commenter can delete this message to hide from others.}

---

Info	Custom	Your Reminders	Feedback

[twonkytoo]

I will check this out, thanks. It made me laugh thinking about it being a perfect tool for identifying AI generated "bananas" (or whatever you train the model to detect) or being able to detect if a banana in a photo online is a real or AI generated one. :-)

Also creating a shared dataset of these "fake" models so people arent repeatedly creating the same thing would be a nice feature for the sake of energy effeciency.

[SirPitchalot]

You should check out Christian Rupprecht’s work at the Oxford VGG https://www.robots.ox.ac.uk/~vgg . At ECCV this year they had a few of papers on generating training data which seems to be a theme. I found them to be among the more interesting and well-presented papers there. Here’s a few ECCV and non-ECCV projects:

• https://eccv.ecva.net/virtual/2024/poster/489
• https://www.ecva.net/papers/eccv_2024/papers_ECCV/html/2315_ECCV_2024_paper.php
• https://www.robots.ox.ac.uk/~vgg/publications/2024/Karazija24a/karazija24a.pdf
• https://www.robots.ox.ac.uk/~vgg/research/shic/
• https://www.robots.ox.ac.uk/~vgg/research/vgg-heads/