微信公众号:OpenCV开发者联盟
关注获取更多计算机视觉与深度学习知识
应用说明
模型介绍
输入格式:
NCHW = 1x3x640x640
输出格式:
1x6x8400, xyhw+类别2 = 6,20x20+40x40+80x80=8400
模型推理基于OpenVINO框架,CPU即可运行。
界面设计
界面支持模型相关得参数设置、输入图像路径选择与设置,相关界面设计如下:
代码实现与演示
# 初始化APP实例
import platform
app = QtWidgets.QApplication(sys.argv)
if 'Windows' == platform.system():
app.setStyle('Windows')
# 初始化桌面容器
main_win = QtWidgets.QMainWindow()
# 设置APP窗口名称
main_win.setWindowTitle("安全帽测检测 - 演示版本")
# 初始化内容面板
content_panel = YOLOv8InferPanel()
# 设置窗口大小
main_win.setMinimumSize(1340, 910)
main_win.setCentralWidget(content_panel)
# 请求显示
main_win.show()
# 加载窗口并启动App
app.exec()
推荐阅读
三行代码实现 TensorRT8.6 C++ 深度学习模型部署