开始训练时,cpu占用率过高,训练比较慢,原因在于cuda与tensorflow版本不匹配,而且没有安装cuDNN
Tensorflow官网对应表如下
重新安装cuda与对应版本的tensorflow和cuDNN解决问题,训练时间大大缩短,10epoch训练完毕大约5min
运行test_on_single_image.py得到推理结果