How to obtain and load a good initial data set for object localization?

Question

I'm looking for a good data set for training a CNN based network to do object localization (i.e. a data set with class labels and bounding box data).

What is a good initial data set to use?

How can I obtain and load all the data (images, labels, bounding boxes)?

I'm using python, so a python-friendly data source is ideal.

Vivek Kalyanarangan · Answer

The ones that are available out of the box -

Open Images Dataset (Object Detection Track)
MS COCO dataset
Indoor Objects Detection
ILSVRC dataset

Create your own

Search for the object of interest on Google Images (like bikes, cars etc.)
Use the Fatkun Chrome browser plugin to download all images at once
Use LabelImg to annotate and generate a custom dataset
Use pre-trained models like SSD MobileNet to re-train your own models

Might not be an exhaustive one, but it will definitely get you started. Cheers!

How to obtain and load a good initial data set for object localization?

One Answer

Add your own answers!

Ask a Question