该博客演示了如何使用TensorFlow的预训练模型进行目标检测，并将检测结果存入MySQL数据库。通过定义`detection_to_database`函数连接MySQL，根据帧编号检查检测状态，更新或插入数据。同时，加载模型数据并执行检测，使用`Detection`函数在图片上可视化检测结果，并调用`detection_to_database`将结果存储到数据库。
#!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是否已经被检测过，若是，则更新，若否，
最低0.47元/天 解锁文章
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
