首先安装依赖的包pip install -r requirements.txt 需要注意的是，这里使用的是tensorflow 2.0, 需要使用较新的cuda和cudnn。我的环境使用的是cuda 10 以及cudnn 7.5 。供大家参考。

需要在congig.yaml文件中配置好相关参数，包括数据集名称、路径，所需的数据预处理操作以及训练使用的模型。

python make_dataset.py创建tfrecord文件，然后执行python training.py开始训练。