Clear Filters
Clear Filters

Importing CelebA dataset (deep learning) into MATLAB for use with YOLO

7 views (last 30 days)
Following the example on the MATLAB website for object detection using YOLO V3, I'd like to replace the dataset provided with the example (vehicle Dataset) with a faces dataset called CelebA.
Here's a link to the dataset on kaggle:
And a link to the Yolo V3 example:
I've downloaded the images of the dataset as well as a list_bbox_celeba.csv file, which, as its name suggests, contains bounding box data for the images.
How can I import this dataset (images + bounding box data) into MATLAB so that I could use the Yolo V3 example code for face detection instead of vehicle detection?

Answers (1)

Mahesh Taparia
Mahesh Taparia on 13 Oct 2021
The training data can have 2 formats either tabular or in the form of datastore as given here. In your case, create a table with file/ image path in 1st column and remaining columns with box labels as indicated in the above link. So somedata preprocessing is required before passing to an object detector.
Hope it will help!




Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!