MLDatasetBuilder 0.0.9

MLDatasetBuilder is a python package which is helping to prepare the image for your ML dataset.

MLDatasetBuilder

MLDatasetBuilder is a python package that is helping to prepare images for your ML dataset. Whenever we begin a machine learning project, the first thing that we need is a dataset. Datasets will be the pillar of the training model. You can build the dataset either automatically or manually.

Author: Karthick Nagarajan

Email: [e-mail]

Installation

We can install MLDatasetBuilder package using this command

pip install MLDatasetBuilder

Available Operations

1. Remove unwanted format images and Rename image files

PrepareImage(folder_name, image_name)
#PrepareImage('images', 'dog')

2. Extract images from video file

ExtractImages(video_path, filename, framesize)
#ExtractImages('video.mp4', 'frame', 10)

Remove unwanted format images and Rename image files

we can get images from Google. Using the Download All Images browser extension we can easily get images in a few minutes. You can check out here for more details about this extension!

Once you have downloaded the images using this extension, you can create a python file called test_python_file.py the same directory as below.

download_image_folder/
   _14e839ba-9691-11ea-a968-2ed746e9a968.jpg
   5e5f7af12600004018b602c0.jpeg
   A471529_Alice_b-1.jpg
   image1.png
   image2.png
   ...
test_python_file.py

Inside the images folder, you can see lots of png images and random filenames.

PrepareImage(folder_name, image_name)
#PrepareImage('images', 'dog')

download_image_folder/
   dog_0.jpg
   dog_1.jpg
   dog_2.jpg
   dog_3.png
   dog_4.png
   ...
test.py

This process very helps to annotate your images while labeling. And of course, it will be like one of the standardized things.

Extract images from video file

download_image_folder/
video.mp4
test.py

ExtractImages(video_path, folder_name, framesize)
#ExtractImages('video.mp4', 'frame', 10)
ExtractImages(video_path, folder_name)
#ExtractImages('video.mp4', 'frame')

download_image_folder/
dog/
   dog_0.jpg
   dog_1.jpg
   dog_2.jpg
   dog_3.png
   dog_4.png
   ...
dog.mp4
test.py

Contact

MLDatasetBuilder was created by Karthick Nagarajan. Feel free to reach out on Twitter or through Email!