H-xie / MNIST2TFRecord

This is a simple way to convert original mnist to TFRecord for a geater effective.

MNIST2TFRecord

This is a simple way to convert original mnist to TFRecord for a geater effective.

Preparation

Some packages is need

pip3 install tensorflow numpy opencv-python progressbar

Usage

1. Need to download and unzip mnist by

python3 download_data.py

2. Run the script to generate tfrecord.

python3 mnist2tfrecord.py

This contains several default options. -h will show all infomation.

Reference

• download_data.py is directly cited from naturomics/CapsNet-Tensorflow

• mnist is a popular handwritten digits dataset.