MySQL数据库连接YOLOv5模型,将yolov5目标检测到的结果保存到mysql数据库中
时间: 2024-03-04 14:48:10 浏览: 128
可以通过Python中的MySQL Connector将YOLOv5模型检测到的结果保存到MySQL数据库中。以下是一个简单的示例代码:
```python
import mysql.connector
# 连接MySQL数据库
mydb = mysql.connector.connect(
host="localhost",
user="yourusername",
password="yourpassword",
database="yourdatabase"
)
# 创建一个游标对象
mycursor = mydb.cursor()
# 创建一个表来存储检测结果
mycursor.execute("CREATE TABLE detection_results (id INT AUTO_INCREMENT PRIMARY KEY, object_name VARCHAR(255), confidence FLOAT)")
# 模拟YOLOv5检测到的结果
results = [("person", 0.95), ("car", 0.85), ("dog", 0.80)]
# 将结果插入到数据库中
for result in results:
sql = "INSERT INTO detection_results (object_name, confidence) VALUES (%s, %s)"
val = (result[0], result[1])
mycursor.execute(sql, val)
# 确保更改被提交到数据库
mydb.commit()
print(mycursor.rowcount, "记录已插入。")
```
在这个示例中,我们首先连接到MySQL数据库,然后创建一个名为"detection_results"的表来存储检测结果。然后,我们模拟YOLOv5检测到的结果并将它们插入到表中。最后,我们确保提交更改并输出插入的记录数。
你可以根据实际需求修改表和数据的结构。
阅读全文