• Ubuntu 14.04.5 LTS
• Python 2.7.6
• Tensorflow 1.4.0-rc1

TFRecord generation:

• DataGenTFRecord.py From raw data generation TFRecord

Read data:

• cifar_input.py : cifar10 or cifar100 input
• sketchy_input.py sketchy input

Model build:

• resnet_model.py : build ResNet Model

Train Model:

• resnet_main.py : train model

Sketchy dataset

• python DataGenTFRecord.py

• ./run.sh

Note:

• The picture has nothing to do with the content!