search

apple / turicreate

Turi Create simplifies the development of custom machine learning models.
BSD 3-Clause "New" or "Revised" License
11.19k stars 1.14k forks source link

Data Augmentation for Image Classifier #1791

Open nseidl opened 5 years ago

nseidl commented 5 years ago

Are there any plans to allow for custom data augmentation (scale, flip, rotate, etc.) through TuriCreate for the turicreate.image_classifier?

If not, are there any recommended python libraries for doing this on images, possibly even working with SFrame?

hoytak commented 5 years ago

Hello @nseidl,

Thanks for the feature request -- yes, this is great!

One way that I've personally used for doing data augmentations is to use ImageMagick wrapped in a script. It's not ideal, but it works. Check out https://github.com/hoytak/diabetic-retinopathy-code for some code to do this.

bamwani commented 4 years ago

@hoytak Hello, any updates?

jamois commented 4 years ago

+1 to the above. Wondering if augmentation (specifically rotating images) is available yet for building image classifiers. Thx!