A image-data-set generator for artificial intelligence training purpose.
How it works: Basicly it walk through a root folder and get all the sub folder as a class name then it create a corresponding folder with the same class name to save csv files using pillow to read imagefiles and transform into rgb data saved in numpy matrix then save the matrix as a csv file, when loading a csv it tranlate the matrix back with some dataset generator functions, it can devide the whole set into train set, validate set and test set.And it can create batches of train set for SGD or other training method.
By Celestial Tech Co. Copyright © 2015-2017 For more check : www.tianshicangxie.com