A simple and extensible project based on TensorFlow-Slim image classification model library
A simple and extensible project based on TensorFlow-Slim image classification model library for
Download and generate mnist, fashion and cifar datasets by:
cd data/xxx/
python3 download_and_convert_xxx.py
For imagenet dataset, please follow the instructions in
an automated script,
and change the destinations in source/datasets/dataset_factory
to the converted tf.record file locations
Change or new an option file in source/options/
, there are some templates for beginners.
Change the configuration in option file:
gpu_list = [2]
gpu_list = [1,3]
gpu_list = []
Change or new a model file in source/nets/
, and register it in nets_factory.py
You can also modify and add other components, such as preprocessing, dataset. Don't forget to register them in individual factory files.
In case the option file name in source/options/
is your_options.py
cd source/
python3 top.py -o your_options
The training log files will be printed and saved in log/time_your_title.txt
You can check the training details in log files, and derive statistics for drawing curves.
And the model will be saved in model/time_your_title
if you have configured the saving parameters.
完成下列实验,提交实验报告,内容包括:
使用MLP训练mnist,/options/mlp_mnist.py
记录:
使用LeNet训练fashion,/options/lenet_fashion.py
记录:
/options/resnet_cifar.py
中的设置