顾名思义，视频字幕提取神器这个就是用来提取视频中字幕区域的文字的，利用了百度识字接口和cv2 Python语言来实现的。某些职业，某些人用得上，比如看视频记笔记，或者摘抄视频中的文案内容等等。

之前一直在用其他的软件，可惜识别精准度不高，视频字幕提取神器明显好用多了。

源码

# encoding:utf-8

import requests,time,cv2

import base64,copy

from PIL import Image

class Baiduorc():

def __init__(self,AK,SK,MS=1):

self.AK=AK

self.Sk=SK

self.MS = MS

host = f’https://aip.baidubce.com/oauth/2.0/token?grant_type=client_credentials&client_id={AK}&client_secret={SK}‘

try:

response = requests.get(host)

if response:

access_token = response.json()[‘access_token’]

self.access_token = access_token

else:

self.access_token = None

except:

self.access_token = None

#获取鉴权接口,就是身份验证…

def get_text(self,imagePath):

”’

识别文字

:param imagePath: 图片地址

:return: 返回结果

”’

if self.MS==1:

request_url = “https://aip.baidubce.com/rest/2.0/ocr/v1/general_basic”

else:

request_url = “https://aip.baidubce.com/rest/2.0/ocr/v1/accurate_basic”

# 二进制方式打开图片文件

with open(imagePath,‘rb’)as f:

img=f.read()

img = base64.b64encode(img)

params = {“image”: img}

access_token = self.access_token

request_url = request_url + “?access_token=” + access_token

headers = {‘content-type’: ‘application/x-www-form-urlencoded’}

response = requests.post(request_url, data=params, headers=headers)

if response:

 

return response.json()

else:

return None

 

if __name__ == “__main__”:

print(‘高精度的接口次数有限,请自己去注册[url]https://cloud.baidu.com/product/ocr_general’[/url])

MS=int(input(“请选择调用精度:1普通识别,2高精度识别\n”))

mp4Name=input(‘请输入你的视频文件的完整目录和文件名:默认为 下载.mp4\n’)

if mp4Name==None or mp4Name==”:

mp4Name=‘下载.mp4’

rectangle= input(‘请输入字幕的矩形范围格式为:x1,y1,x2,y2 默认为:36,292,598,330 如果输入0则为全屏 \n矩形坐标可以先默认,然后运行显示起来用截图工具把字幕矩阵区域的坐标搞到来!\n’)

if rectangle==None or rectangle==”:

rectangle=[36,292,598,330]

elif rectangle==‘0’:

pass

else:

rectangle =rectangle.split(‘,’)

timeS=input(‘请输入视频截图间隔多少帧,默认为60\n’)# 视频帧计数间隔频率

if timeS==None or timeS==”:

timeS=60

else:

timeS=int(timeS)

#百度识字

# client_id 为官网获取的AK， client_secret 为官网获取的SK

with open(‘百度ak和sk.txt’,‘r’)as f:

ak_and_sk=f.read()

ak_and_sk=ak_and_sk.split(‘\n’)

 

AK = ak_and_sk[0].split(‘”‘)[1]

 

SK = ak_and_sk[1].split(‘”‘)[1]

print(‘百度ak和sk’,AK, SK)

bd = Baiduorc(AK=AK, SK=SK,MS=MS)

imagePath = “1.png”#临时存放目录

c = 0

cap = cv2.VideoCapture(mp4Name)

last_text=”

now_text=”

text=”

while (cap.isOpened()):

c = c + 1

ret, frame = cap.read()

frame

if c%timeS==0: # c除尽timeS时取帧保存，图片，即隔timeS保存一次图片

#cv2.imwrite(imagePath,frame)

#img = cv2.im(frame)

#37,297,534,336

if rectangle==‘0’:#全屏识别

cv2.imwrite(imagePath, frame)

else:

try:

frame_jq=frame[rectangle[1]:rectangle[3], rectangle[0]:rectangle[2]]

cv2.imwrite(imagePath, frame_jq)

except:

break

jstext = bd.get_text(imagePath)

try:

now_text=jstext[‘words_result’][0][‘words’]

if now_text!=last_text:

text=text+now_text+‘\n\r’

print(jstext[‘words_result’][0][‘words’])

last_text=copy.copy(now_text)

except:

pass

cv2.imshow(‘image’, frame)

k = cv2.waitKey(20)

# q键退出

if (k & 0xff == ord(‘q’)):

 

break

 

cap.release()

cv2.destroyAllWindows()

with open(‘test.txt’,‘w’)as f:

f.write(text)

input(‘完成’)

蓝奏云：https://lanren.lanzoux.com/b00u5c9wj 密码:e2ae