在python中通过obs websocket获取截图后想要将其转化为opencv可以操作的格式,但是试了很多办法都失败了
OBS28.0.1更新后自带的obs-websocket 5
import obsws_python as obs
import numpy as np
import base64
import cv2
import sys
cl = obs.ReqClient(host='localhost', port=4444, password="111111") #连接到websocket服务器
image_file_path = sys.path[0] + "\Screenshot.png"
source = "Video Capture Device" #这里填添加到OBS中的源的名字,显示在OBS主界面来源中的名字
Screenshot = cl.get_source_screenshot(source,"jpg",None,None,100) #通过websocket从obs请求对源进行截图
Screenshot = Screenshot.image_data
img = base64.b64decode(Screenshot)
img_array = np.fromstring(img, np.uint8) # 转换np序列
print('numpy: ', img_array.shape)
cv2.imshow("img", img_array)
cv2.waitKey(0)
在b64decode处报错:Incorrect padding
websocket官方说明书里面说这个image_data是“base64 encoded”,于是找了很多种base64转np array的方法,都以失败告终
修改get_source_screenshot的图片格式,会出现不同的报错
在python中通过obs websocket获取截图后想要将其转化为opencv可以操作的格式
但是不希望用save_source_screenshot保存图片后读取的方式,因为我可能会要求程序循环获取截图,这样会严重降低硬盘寿命。
可能需要大家安装一下OBS进行调试,如果能直接看到cl.get_source_screenshot获取的数据应该能更快找到解决方法吧
OBS下载:
https://obsproject.com/
obs-websocket 5 已经内置在该版本中,说明书:https://github.com/obsproject/obs-websocket/blob/master/docs/generated/protocol.md#getsourcescreenshot
OBS安装好后:
在OBS主界面上方 工具 - obs-websocket设置 - 开启WebSocket服务器,开启鉴权,设置服务器端口(int)和服务器密码(string)并将端口和密码填写到cl = obs.ReqClient(host='localhost', port=端口号, password="密码")
随便加个源,把自己设置的源名称填写到source中。
python中base64串的长度为4的整数倍,因此长度不为4整数倍的base64串需要用"='补全
代码稍微修改了下,你再试试
import obsws_python as obs
import numpy as np
import base64
import cv2
import sys
cl = obs.ReqClient(host='localhost', port=4444, password="931122")
image_file_path = sys.path[0] + "\Screenshot.png"
source = "Video Capture Device"
Screenshot = cl.get_source_screenshot(source,"jpg",None,None,100)
Screenshot = Screenshot.image_data
#长度不为4整数倍的base64串需要用"='补全
num = len(Screenshot)%4
if num != 0:
Screenshot=Screenshot + '=' * (4-num)
img = base64.b64decode(Screenshot)
img_array = np.fromstring(img, np.uint8) # 转换np序列
print('numpy: ', img_array.shape)
cv2.imshow("img", img_array)
cv2.waitKey(0)
需要解码:cv.imdecode(nparr, 1) #转为HWC,有用记得采纳
nparr = np.frombuffer(img_data, np.uint8)
img_decode = cv.imdecode(nparr, 1) #转为HWC
cv.imshow('', img_decode)
cv.waitKey()
cv.destroyAllWindows()
问题应该是在转码的地方,必要时可远程协助一下。
解码问题。楼上老哥的base64不足4位添加==的基础上面,将你的
img_array = np.fromstring(img, np.uint8) # 转换np序列
换成
img_array = np.array(bytearray(img), dtype=np.uint8) # 转换np序列
截图是可以截图,但是有大部分数据截取的不对,只有左边那一列白色部分的截图是对的,估计是我推流啥的设置不对吧
朋友,你的open cv版本可能不对,它是?
根据你上面的报错截图,你的numpy是一维格式的,但是698506这个数好像除不尽3,也就是说reshape成RGB图像给opencv展示不了,你这base64是必要的嘛?