linux 小程序开发日记-6
这一篇是该文章的最后第二集
最后一篇博客地址：https://blog.csdn.net/Liuchengzhizhi/article/details/123692365
B站视频：https://www.bilibili.com/video/BV1rZ4y1B7t8?share_source=copy_web
源码：https://gitee.com/wx_b915676bb6/yolo-pyqt.git
今天主要解决的问题是python连接数据库的问题，并同时进行数据库的录入。
- ubuntu 安装mysql和其可视化工具
- 连接数据库
- 处理获取的lable list，将其可以录入数据库
下面先附上我自己的视频，主要内容是讲解了一下上面任务的代码
https://www.bilibili.com/video/BV1WY411n79M?share_source=copy_web
第一步 安装mysql及可视化工具
安装我就附上我几篇参考的博客
安装mysql
https://www.cnblogs.com/2020javamianshibaodian/p/12920243.html
安装 mysql workbench
这个我是去官网下载的deb包，然后通过dpkg 命令进行安装
官网-》 https://dev.mysql.com/downloads/workbench/
修复依赖包broken问题
sudo apt --fix-broken install
第二步连接数据库
这里我自己写了一个小的python文件，用于连接数据库
代码如下
# @Author: Cjuicy
# @Date: 2022-03-23 09:37:21
# @Last Modified by: Cjuicys
# @Last Modified time: 2022-03-23 09:37:21
from logging import root
import pymysql
dbhost='localhost'
dbuser='root'
dbpass='12345678'
dbname='order_info'
try:
db=pymysql.connect(host=dbhost,user=dbuser,password=dbpass,database=dbname)
print("数据库连接成功")
except pymysql.Error as e:
print("数据库连接失败："+str(e))
def insert_order_info(order_info):
# 对有字符串的内容，添加双引号，这样才能够读取 -1是应为要插入时间，时间不能加双引号
for i in range(len(order_info)-1):
if order_info[i] != 'null':
order_info[i] = "'" + order_info[i] + "'"
order_info = ','.join(order_info)
#数据库文字
sql = "INSERT INTO `order_info`.`OrderInfo`VALUES(%s);"%(order_info)
print(sql)
cursor = db.cursor()
cursor.execute(sql)
db.commit
print('插入成功')
#显示所有数据库字段
# select = "SELECT * FROM order_info.OrderInfo;"
# cursor.execute(select)
# alldata = cursor.fetchall()
第三步 对获取的lable list 进行处理，使其能够适用于sql语句
代码文件是detect_logical
主要修改的代码
'''初始化所有槽函数'''
def slot_init(self):
self.ui.button_confirm.clicked.connect(self.button_video_stop) #若该按键被点击，则调用button_confirm()
self.ui.button_settle_accounts.clicked.connect(self.insert_order)
self.timer_video.timeout.connect(self.show_video_frame) # 定时器超时，将槽绑定至show_video_frame
# 处理label集的函数
def set_labels(self ,labels ):
self.ui.list_show.clearContents() #对表格先进行一个清空
labels = pd.value_counts(labels) #对标签数组进行重复个数统计
total_price = 0
order_sql = []
if len(labels.index.values) != 0:
labels_sum = len(labels.index.values) #获取标签总数
for i in range(labels_sum):
label_num = str(labels.values[i]) #获取对应标签的个数
self.ui.list_show.setItem(i,0,QTableWidgetItem(labels.index.values[i])) #添加对应标签名字
self.ui.list_show.setItem(i,1,QTableWidgetItem(label_num)) #添加对应标签个数
#记录数据库要插入的字符
order_sql.append(str(labels.index.values[i]))
order_sql.append(str(label_num))
# 判断字典内是否有该key,有添加对应价钱数据，没有则选择默认值（99元）
if labels.index.values[i] in self.price_info :
label_price = str(self.price_info[labels.index.values[i]])
self.ui.list_show.setItem(i,2,QTableWidgetItem(label_price))
# 计算总价
total_price = total_price + (float(label_price) * int(label_num))
else :
self.ui.list_show.setItem(i,2,QTableWidgetItem("99"))
# 计算总价
total_price = total_price + (float(99) * int(label_num))
#设置数据库内容
if i+1>=labels_sum: #判断是不是遍历到了标签最后
while(i<5): #循环，添加剩余的例如数据库的内容
order_sql.append('null')
order_sql.append('null')
i = i+1
account = "总价： "+ str(total_price)
self.ui.label_account.setText(account)
#设置数据库
order_sql.insert(0,str(total_price)) #插入总价钱
order_sql.insert(0,"null") #插入id变量
order_sql.append("CURRENT_TIMESTAMP") #插入当前时间
# order_sql = ','.join(order_sql) #将数组转化为字符串
print(order_sql)
self.order_info = order_sql
# self.order_info = order_sql.replace(" ","_") #将空格替换为_（不换数据库报错）
#插入数据
def insert_order(self):
food_sql.insert_order_info(self.order_info)
所有代码如下
#
# @Author: Cjuicy
# @Date: 2022-03-21 16:00:23
# @Last Modified by: Cjuicys
# @Last Modified time: 2022-03-21 16:00:23
#
from cProfile import label
from pickle import TRUE
import sys
from unittest import result
import cv2
import time
import argparse
import random
import torch
import numpy as np
import pandas as pd
import torch.backends.cudnn as cudnn
from PyQt5 import QtCore, QtGui, QtWidgets
from PyQt5.QtCore import *
from PyQt5.QtGui import *
from PyQt5.QtWidgets import *
from utils.torch_utils import select_device
from models.experimental import attempt_load
from utils.general import check_img_size, non_max_suppression, scale_coords
from utils.datasets import letterbox
from utils.plots import plot_one_box2
from utils.torch_utils import select_device, load_classifier, time_synchronized
from ui.food_ui import Ui_MainWindow # 导入detect_ui的界面
from utils import price_utils ,food_sql
class Ui_Logic_Window(QtWidgets.QWidget):
def __init__(self,parent=None):
super().__init__(parent) #父类的构造函数
self.timer_video = QtCore.QTimer() #定义定时器，用于控制显示视频的帧率
self.cap = cv2.VideoCapture() #视频流
self.ui = Ui_MainWindow()
self.price_info = price_utils.get_price_info() # 初始化 获取表格价钱信息（应为量少，可以这么做，安全等其他因素，应该走数据库）
layout_main = self.ui.set_ui() #初始化页面布局，并获取所有的页面布局
'''将总布局作为参数传入下面函数'''
self.setLayout(layout_main) #到这步才会显示所有控件
self.slot_init() #初始化槽函数
self.num_stop = 1 # 暂停与播放辅助信号，note：通过奇偶来控制暂停与播放
self.output_folder = 'output/'
self.vid_writer = None
self.order_info = ''
# 权重初始文件名
#模型路径
self.openfile_name_model = "./weights/yolov5s.pt"
self.model_init()
# 打开摄像头
self.button_camera_open()
#将结账按钮触发禁用
self.ui.button_settle_accounts.setDisabled(True)
'''初始化所有槽函数'''
def slot_init(self):
self.ui.button_confirm.clicked.connect(self.button_video_stop) #若该按键被点击，则调用button_confirm()
self.ui.button_settle_accounts.clicked.connect(self.insert_order)
self.timer_video.timeout.connect(self.show_video_frame) # 定时器超时，将槽绑定至show_video_frame
# ========================================================
# 加载相关参数，并初始化模型
def model_init(self):
# 模型相关参数配置
parser = argparse.ArgumentParser()
parser.add_argument('--weights', nargs='+', type=str, default='weights/yolov5s6.pt', help='model.pt path(s)')
parser.add_argument('--source', type=str, default='data/images', help='source') # file/folder, 0 for webcam
parser.add_argument('--img-size', type=int, default=640, help='inference size (pixels)')
parser.add_argument('--conf-thres', type=float, default=0.25, help='object confidence threshold')
parser.add_argument('--iou-thres', type=float, default=0.45, help='IOU threshold for NMS')
parser.add_argument('--device', default='0', help='cuda device, i.e. 0 or 0,1,2,3 or cpu')
parser.add_argument('--view-img', action='store_true', help='display results')
parser.add_argument('--save-txt', action='store_true', help='save results to *.txt')
parser.add_argument('--save-conf', action='store_true', help='save confidences in --save-txt labels')
parser.add_argument('--nosave', action='store_true', help='do not save images/videos')
parser.add_argument('--classes', nargs='+', type=int, help='filter by class: --class 0, or --class 0 2 3')
parser.add_argument('--agnostic-nms', action='store_true', help='class-agnostic NMS')
parser.add_argument('--augment', action='store_true', help='augmented inference')
parser.add_argument('--update', action='store_true', help='update all models')
parser.add_argument('--project', default='runs/detect', help='save results to project/name')
parser.add_argument('--name', default='exp', help='save results to project/name')
parser.add_argument('--exist-ok', action='store_true', help='existing project/name ok, do not increment')
self.opt = parser.parse_args()
print(self.opt)
# 默认使用opt中的设置（权重等）来对模型进行初始化
source, weights, view_img, save_txt, imgsz = self.opt.source, self.opt.weights, self.opt.view_img, self.opt.save_txt, self.opt.img_size
# 若openfile_name_model不为空，则使用此权重进行初始化
if self.openfile_name_model:
weights = self.openfile_name_model
print("Using button choose model")
self.device = select_device(self.opt.device)
self.half = self.device.type != 'cpu' # half precision only supported on CUDA
cudnn.benchmark = True
# Load model
self.model = attempt_load(weights, map_location=self.device) # load FP32 model
stride = int(self.model.stride.max()) # model stride
self.imgsz = check_img_size(imgsz, s=stride) # check img_size
if self.half:
self.model.half() # to FP16
# Second-stage classifier
classify = False
if classify:
modelc = load_classifier(name='resnet101', n=2) # initialize
modelc.load_state_dict(torch.load('weights/resnet101.pt', map_location=self.device)['model']).to(self.device).eval()
# Get names and colors
self.names = self.model.module.names if hasattr(self.model, 'module') else self.model.names
self.colors = [[random.randint(0, 255) for _ in range(3)] for _ in self.names]
print("model initial done")
# 设置提示框
QtWidgets.QMessageBox.information(self, u"Notice", u"模型加载完成", buttons=QtWidgets.QMessageBox.Ok,
defaultButton=QtWidgets.QMessageBox.Ok)
# 目标检测
def detect(self, name_list, img):
'''
:param name_list: 文件名列表
:param img: 待检测图片
:return: info_show:检测输出的文字信息
'''
showimg = img
with torch.no_grad():
img = letterbox(img, new_shape=self.opt.img_size)[0]
# Convert
img = img[:, :, ::-1].transpose(2, 0, 1) # BGR to RGB, to 3x416x416
img = np.ascontiguousarray(img)
img = torch.from_numpy(img).to(self.device)
img = img.half() if self.half else img.float() # uint8 to fp16/32
img /= 255.0 # 0 - 255 to 0.0 - 1.0
if img.ndimension() == 3:
img = img.unsqueeze(0)
# Inference
pred = self.model(img, augment=self.opt.augment)[0]
# Apply NMS
pred = non_max_suppression(pred, self.opt.conf_thres, self.opt.iou_thres, classes=self.opt.classes,
agnostic=self.opt.agnostic_nms)
info_show = ""
# 定义标签变量名
label_names = ""
# Process detections
for i, det in enumerate(pred):
if det is not None and len(det):
# Rescale boxes from img_size to im0 size
det[:, :4] = scale_coords(img.shape[2:], det[:, :4], showimg.shape).round()
for *xyxy, conf, cls in reversed(det):
label = '%s %.2f' % (self.names[int(cls)], conf)
name_list.append(self.names[int(cls)])
single_info = plot_one_box2(xyxy, showimg, label=label, color=self.colors[int(cls)], line_thickness=2)
# print(single_info)
info_show = info_show + single_info + "\n"
# 新添加的标签信息,提交给pyqt5 处理
label_names = label_names+","+self.names[int(cls)]
# 转化为数组
label_names = label_names.split(",")
# 去掉空值
label_names = [i for i in label_names if(len(str(i))!=0)]
return info_show , label_names
def set_video_name_and_path(self):
# 获取当前系统时间，作为img和video的文件名
now = time.strftime("%Y-%m-%d-%H-%M-%S", time.localtime(time.time()))
# if vid_cap: # video
fps = self.cap.get(cv2.CAP_PROP_FPS)
w = int(self.cap.get(cv2.CAP_PROP_FRAME_WIDTH))
h = int(self.cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
# 视频检测结果存储位置
save_path = self.output_folder + 'video_output/' + now + '.mp4'
return fps, w, h, save_path
# 打开摄像头检测
def button_camera_open(self):
print("Open camera to detect")
# 设置使用的摄像头序号，系统自带为0
camera_num = 0
# 打开摄像头
self.cap = cv2.VideoCapture(camera_num)
# 判断摄像头是否处于打开状态
bool_open =self.cap.open(camera_num)
if not bool_open:
QtWidgets.QMessageBox.warning(self, u"Warning", u"打开摄像头失败", buttons=QtWidgets.QMessageBox.Ok,
defaultButton=QtWidgets.QMessageBox.Ok)
else:
fps, w, h, save_path = self.set_video_name_and_path()
# fps = 5 # 控制摄像头检测下的fps，Note：保存的视频，播放速度有点快，我只是粗暴的调整了FPS
self.vid_writer = cv2.VideoWriter(save_path, cv2.VideoWriter_fourcc(*'mp4v'), fps, (w, h))
self.timer_video.start(30)
# 定义视频帧显示操作
def show_video_frame(self):
name_list = []
flag, img = self.cap.read()
if img is not None:
info_show , label_names = self.detect(name_list, img) # 检测结果写入到原始img上
self.vid_writer.write(img) # 检测结果写入视频
print(info_show)
#处理lable并显示在GUI上
self.set_labels(label_names)
show = cv2.resize(img, (640, 480)) # 直接将原始img上的检测结果进行显示
self.result = cv2.cvtColor(show, cv2.COLOR_BGR2RGB)
showImage = QtGui.QImage(self.result.data, self.result.shape[1], self.result.shape[0],
QtGui.QImage.Format_RGB888)
# self.ui.label.setPixmap(QtGui.QPixmap.fromImage(showImage))
self.ui.label_show_camera.setPixmap(QtGui.QPixmap.fromImage(showImage)) #往显示视频的Label里 显示QImage
# self.ui.label.setScaledContents(True) # 设置图像自适应界面大小
else:
self.timer_video.stop()
# 读写结束，释放资源
self.cap.release() # 释放video_capture资源
self.vid_writer.release() # 释放video_writer资源
self.ui.label.clear()
# 视频帧显示期间，禁用结账按键功能
self.ui.button_settle_accounts.setDisabled(True)
# 暂停与继续检测
def button_video_stop(self):
self.timer_video.blockSignals(False)
# 暂停检测
# 若QTimer已经触发，且激活
if self.timer_video.isActive() == True and self.num_stop%2 == 1:
self.ui.button_confirm.setText(u'重新确认') # 当前状态为暂停状态
self.num_stop = self.num_stop + 1 # 调整标记信号为偶数
self.timer_video.blockSignals(True)
# 启动结账按钮
self.ui.button_settle_accounts.setDisabled(False)
# 继续检测
else:
self.num_stop = self.num_stop + 1
self.ui.button_confirm.setText(u'确认')
self.ui.button_settle_accounts.setDisabled(True)
# 处理label集的函数
def set_labels(self ,labels ):
self.ui.list_show.clearContents() #对表格先进行一个清空
labels = pd.value_counts(labels) #对标签数组进行重复个数统计
total_price = 0
order_sql = []
if len(labels.index.values) != 0:
labels_sum = len(labels.index.values) #获取标签总数
for i in range(labels_sum):
label_num = str(labels.values[i]) #获取对应标签的个数
self.ui.list_show.setItem(i,0,QTableWidgetItem(labels.index.values[i])) #添加对应标签名字
self.ui.list_show.setItem(i,1,QTableWidgetItem(label_num)) #添加对应标签个数
#记录数据库要插入的字符
order_sql.append(str(labels.index.values[i]))
order_sql.append(str(label_num))
# 判断字典内是否有该key,有添加对应价钱数据，没有则选择默认值（99元）
if labels.index.values[i] in self.price_info :
label_price = str(self.price_info[labels.index.values[i]])
self.ui.list_show.setItem(i,2,QTableWidgetItem(label_price))
# 计算总价
total_price = total_price + (float(label_price) * int(label_num))
else :
self.ui.list_show.setItem(i,2,QTableWidgetItem("99"))
# 计算总价
total_price = total_price + (float(99) * int(label_num))
#设置数据库内容
if i+1>=labels_sum: #判断是不是遍历到了标签最后
while(i<5): #循环，添加剩余的例如数据库的内容
order_sql.append('null')
order_sql.append('null')
i = i+1
account = "总价： "+ str(total_price)
self.ui.label_account.setText(account)
#设置数据库
order_sql.insert(0,str(total_price)) #插入总价钱
order_sql.insert(0,"null") #插入id变量
order_sql.append("CURRENT_TIMESTAMP") #插入当前时间
# order_sql = ','.join(order_sql) #将数组转化为字符串
print(order_sql)
self.order_info = order_sql
# self.order_info = order_sql.replace(" ","_") #将空格替换为_（不换数据库报错）
#插入数据
def insert_order(self):
food_sql.insert_order_info(self.order_info)
if __name__ == '__main__':
app = QtWidgets.QApplication(sys.argv)
current_ui =Ui_Logic_Window() #实例化Ui_MainWindow
current_ui.show() #调用ui的show()以显示。同样show()是源于父类QtWidgets.QWidget的
sys.exit(app.exec_())
最后
当我写完这个数据库的一部分之后，我真的很开心，这意味着我的毕设差不多结束啦。
非常开心！！！
最后附上运行效果
1466
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
