diff --git a/README.md b/README.md index 4bc396f..956819f 100644 --- a/README.md +++ b/README.md @@ -1,2 +1,16 @@ # CINIC-10-TFLoader -CINIC-10 Tensorflow Loader + +This repository contains a data loader of "[CINIC-10] (https://github.com/BayesWatch/cinic-10)" dataset for TensorFlow. A training case using TFLearn is given in `residual_network_cinic10.py`. + +CINIC-10 data can be loaded with the line below: +``` +XTrain, YTrain, XVal, YVal, XTest, YTest = cinic10.loadData("/path/to/dataset/folder", oneHot=True) +``` + +This code returns 6 numpy arrays respectively: Training set images, training set labels, validation set images, validation set labels, testing set images, testing set labels. + +You can check `residual_network_cinic10.py` file for an example. + + +## License +The source code is licensed under [MIT License](./LICENSE).