最近工作的项目使用了TensorFlow中的目标检测技术,通过训练自己的样本集得到模型来识别游戏中的物体,在这里总结下. 本文介绍在Windows系统下,使用TensorFlow的object detection API来训练自己的数据集,所用的模型为ssd_mobilenet,当然也可以使用其他模型,包括ssd_inception.faster_rcnn.rfcnn_resnet等,其中,ssd模型在各种模型中性能最好,所以便采用它来进行训练. 配置环境 1. 在GitHub上下载所需的mod…