CINIC-10 Tensorflow Loader
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

19 lines
795 B

6 years ago
# CINIC-10-TFLoader
6 years ago
6 years ago
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`.
6 years ago
CINIC-10 data can be loaded with the line below:
```
XTrain, YTrain, XVal, YVal, XTest, YTest = cinic10.loadData("/path/to/dataset/folder", oneHot=True)
```
6 years ago
`loadData()` method returns 6 numpy arrays: Training set images, training set labels, validation set images, validation set labels, testing set images, testing set labels, respectively.
6 years ago
6 years ago
This script also downloads and extracts dataset files if dataset is not in the specified folder.
6 years ago
You can check `residual_network_cinic10.py` file for an example.
## License
The source code is licensed under [MIT License](./LICENSE).