为什么80%的码农都做不了架构师?>>>
import cv2
import numpyas np
coor_x,coor_y= -1, -1 # 初始值并无意义,只是为了能够使用np.row_stack函数
# """定义视频编码器
# FourCC全称Four-Character Codes,代表四字符代码 (four character code),
# 它是一个32位的标示符,其实就是typedef unsigned int FOURCC;
# 是一种独立标示视频数据流格式的四字符代码。
# 因此cv2.VideoWriter_fourcc()函数的作用是输入四个字符代码即可得到对应的视频编码器。
# """
def video_cut_action():
fourcc= cv2.VideoWriter_fourcc(*'MPEG')#(*'XVID') # 使用XVID编码器
camera= cv2.VideoCapture('E:/test.mp4')#('细胞极体拨动合成视频.wmv') # 从文件读取视频,Todo:只需要修改成自己的视频路径即可进行测试
fps= camera.get(cv2.CAP_PROP_FPS)# 获取视频帧率
print('视频帧率:%d fps' %fps)
# 判断视频是否成功打开
if (camera.isOpened()):
print('视频已打开')
else:
print('视频打开失败!')
# # 测试用,查看视频size
size= (int(camera.get(cv2.CAP_PROP_FRAME_WIDTH)),
int(camera.get(cv2.CAP_PROP_FRAME_HEIGHT)))
print ('视频尺寸:'+repr(size))
coor= np.array([[1,1]])# Todo:初始值并无意义,只是为了能够使用np.row_stack函数
def OnMouseAction(event,x,y,flags,param):
global coor_x,coor_y,coor
if event == cv2.EVENT_LBUTTONDOWN:
print("左键点击")
print("%s" %x,y)
coor_x,coor_y= x ,y
coor_m= [coor_x,coor_y]
coor= np.row_stack((coor,coor_m))
elif event==cv2.EVENT_LBUTTONUP:
cv2.line(img, (coor_x, coor_y), (coor_x, coor_y), (255, 255, 0), 7)
elif event==cv2.EVENT_RBUTTONDOWN:
print("右键点击")
elif flags==cv2.EVENT_FLAG_LBUTTON:
print("左鍵拖曳")
elif event==cv2.EVENT_MBUTTONDOWN:
print("中键点击")
'''
创建回调函数的函数setMouseCallback();
下面把回调函数与OpenCV窗口绑定在一起
'''
grabbed, img= camera.read()# 逐帧采集视频流
cv2.namedWindow('Image')
cv2.setMouseCallback('Image',OnMouseAction)
while(1):
cv2.imshow('Image',img)
k=cv2.waitKey(1)& 0xFF
if k==ord(' '): # 空格退出操作
break
cv2.destroyAllWindows()# 关闭页面
Width_choose= coor[2,0]-coor[1,0]# 选中区域的宽
Height_choose= coor[2, 1]- coor[1, 1]# 选中区域的高
print('.......................width..................height')
# print(Width_choose)
# print(Height_choose)
print("视频选中区域的宽:%d" %Width_choose,'\n'"视频选中区域的高:%d" %Height_choose)
print("111111111111")
out= cv2.VideoWriter('E://video_test//test_new.mp4',fourcc, fps, (Width_choose,Height_choose))# 参数分别是:保存的文件名、编码器、帧率、视频宽高 avi
Video_choose = np.zeros((Width_choose, Height_choose, 3), np.uint8)
# videoWriter = cv2.VideoWriter('E://video_test//test_new.mp4', cv2.VideoWriter_fourcc('m', 'p', '4', 'v'), fps, size)
while True:
grabbed, frame= camera.read()# 逐帧采集视频流
if not grabbed:
break
gray_lwpCV= cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)# 转灰度图
frame_data= np.array(gray_lwpCV)# 每一帧循环存入数组
box_data= frame_data[coor[1,1]:coor[2,1], coor[1,0]:coor[2,0]]# 取矩形目标区域
pixel_sum= np.sum(box_data, axis=1)# 行求和q
x= range(Height_choose)
emptyImage= np.zeros((Width_choose* 10, Height_choose* 2, 3), np.uint8)
Video_choose= frame[coor[1,1]:coor[2,1],coor[1,0]:coor[2,0]]
out.write(Video_choose)
cv2.imshow('Video_choose', Video_choose)
for iin x:
cv2.rectangle(emptyImage, (i*2, (Width_choose-pixel_sum[i]//255)*10), ((i+1)*2, Width_choose*10), (255, 0, 0), 1)
emptyImage = cv2.resize(emptyImage, (320, 240))
lwpCV_box = cv2.rectangle(frame, tuple(coor[1,:]), tuple(coor[2,:]), (0, 255, 0), 2)
cv2.imshow('lwpCVWindow', frame)# 显示采集到的视频流
# videoWriter.write(lwpCV_box) # 将截取到的画面写入“新视频”
# videoWriter = ('lwpCVWindow', frame)
# cv2.imshow('sum', emptyImage) # 显示画出的条形图
key= cv2.waitKey(1)& 0xFF
if key== ord('q'):
break
out.release()
camera.release()
cv2.destroyAllWindows()
def select_action():
cv2.destroyAllWindows()
if __name__== "__main__":
video_cut_action()
作者:LeiLv
链接:https://www.jianshu.com/p/fae269f4c24a
来源:简书
简书著作权归作者所有,任何形式的转载都请联系作者获得授权并注明出处。