ckh2007

data_generator的地方可以用线程优化，我还没改完，明天发出来

作者方便留个联系方式吗

还有个安装时的bug，pip安装会把所有2.2以上的tensorflow版本都装一遍