mysql检测写入_tensorflow利用预训练模型进行目标检测（三）：将检测结果存入mysql数据库...

#!usr/bin/python#-*- coding: utf-8 -*-

importnumpy as npimportmatplotlib

matplotlib.use('Agg')importmatplotlib.pyplotfrom matplotlib importpyplot as pltimportosimporttensorflow as tffrom PIL importImagefrom object_detection.utils importlabel_map_utilfrom object_detection.utils importvisualization_utils as vis_utilimportdatetime#关闭tensorflow警告

importtimeimportMySQLdbimportargparseimportsys

reload(sys)

sys.setdefaultencoding('utf8')

os.environ['TF_CPP_MIN_LOG_LEVEL']='3'detection_graph=tf.Graph()#将detection的结果存入mysql数据库

defdetection_to_database(object_name, frame_num):

conn=MySQLdb.connect(user='root',passwd='****',host='localhost',port=3306,db='rdshare',charset='utf8')

cursor=conn.cursor()#查询目标检测状态表，查看frame_num是否已经被检测过，若是，则更新，若否，