OpenCV实现摄像头图像分类(Python版)

先安装MMEdu库!

MMEdu安装:https://blog.csdn.net/zyl_coder/article/details/132483865

下面的代码请在Jupyter上运行,并自己准备数据集。若模型还未训练,请先在本地训练完模型后再进行模型推理。

python 复制代码
import cv2
capture = cv2.VideoCapture(0)
from MMEdu import MMClassification as cls
model = cls(backbone='MobileNet')
checkpoint = '   '   #填写最佳权重文件
model.load_checkpoint( checkpoint = checkpoint)
while True:
    ret, frame = capture.read()
    frame = cv2.flip(frame,1)
    result = model.fast_inference(image=frame, show=False)
    x = model.print_result(result)
    text = ""
    if x[0]['标签'] == 0:
        text = '   '   #图像类别1
    else:
        text = '   '   #图像类别2
    cv2.putText(frame , text, (200, 100), cv2.FONT_HERSHEY_COMPLEX, 1.0, (100, 200, 200), 5)

    cv2.imshow("video", frame)
    key = cv2.waitKey(50)
    #print(key)
    if key  == ord('q'):
        break
cv2.destroyAllWindows()

注意:1.因为我们是面对摄像头的,所以通过flip对摄像头获得的内容做了左右镜像;

2.通过waitKey这个函数进行获取按下的键(以毫秒为单位,这里就是每50毫秒获取一 次),当按下q键时,关闭视频。


本文内容为小编自己汇总,内容可能会有错误或疏漏,感谢大家的提议!

记得点赞和关注哦~

相关推荐
nbsaas-boot1 小时前
Java 正则表达式白皮书:语法详解、工程实践与常用表达式库
开发语言·python·mysql
仗剑_走天涯1 小时前
基于pytorch.nn模块实现线性模型
人工智能·pytorch·python·深度学习
chao_7891 小时前
二分查找篇——搜索旋转排序数组【LeetCode】两次二分查找
开发语言·数据结构·python·算法·leetcode
风无雨1 小时前
GO 启动 简单服务
开发语言·后端·golang
斯普信专业组2 小时前
Go语言包管理完全指南:从基础到最佳实践
开发语言·后端·golang
gaosushexiangji3 小时前
利用sCMOS科学相机测量激光散射强度
大数据·人工智能·数码相机·计算机视觉
我是苏苏3 小时前
C#基础:Winform桌面开发中窗体之间的数据传递
开发语言·c#
斐波娜娜3 小时前
Maven详解
java·开发语言·maven
小码氓4 小时前
Java填充Word模板
java·开发语言·spring·word
暮鹤筠4 小时前
[C语言初阶]操作符
c语言·开发语言