#!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是否已经被检测过,若是,则更新,若否,