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键时,关闭视频。


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

记得点赞和关注哦~

相关推荐
vvilkim11 分钟前
Nuxt.js 全面测试指南:从单元测试到E2E测试
开发语言·javascript·ecmascript
amazinging25 分钟前
北京-4年功能测试2年空窗-报培训班学测开-第四十四天
python·学习·appium
UrbanJazzerati37 分钟前
Xlwings安装报错:Connection timed out & WinError 32?一招解决你的安装难题!
python
Tipriest_38 分钟前
Python异常类型介绍
开发语言·python·异常
前端付豪43 分钟前
21、用 Python + Pillow 实现「朋友圈海报图生成器」📸(图文合成 + 多模板 + 自动换行)
后端·python
Swift社区1 小时前
Swift 解 LeetCode 321:拼接两个数组中的最大数,贪心 + 合并全解析
开发语言·leetcode·swift
Black_Rock_br1 小时前
语音交互新纪元:Hugging Face LeRobot如何让机器人真正“懂你”
人工智能·计算机视觉·机器人
1900_1 小时前
【论文解读】Referring Camouflaged Object Detection
人工智能·目标检测·计算机视觉
ruan1145141 小时前
Java Lambda 类型推断详解:filter() 方法与 Predicate<? super T>
java·开发语言·spring·stream
猿榜1 小时前
魔改编译-永久解决selenium痕迹(二)
javascript·python