#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Created on Tue Sep 19 10:17:22 2017
@author: vicky
"""
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Created on Mon Sep 18 14:12:01 2017
@author: vicky
"""
import numpy as np
import cv2
import matplotlib.pyplot as plt
import matplotlib.animation as animation
cap1 = cv2.VideoCapture('/Users/vicky/Desktop/第五问视频/4p-c0.mp4')
cap2 = cv2.VideoCapture('/Users/vicky/Desktop/第五问视频/4p-c1.mp4')
cap3 = cv2.VideoCapture('/Users/vicky/Desktop/第五问视频/4p-c2.mp4')
cap4 = cv2.VideoCapture('/Users/vicky/Desktop/第五问视频/4p-c3.mp4')
kernel = cv2.getStructuringElement(cv2.MORPH_ELLIPSE,(5,5))#形态学运算,膨胀
fgbg = cv2.createBackgroundSubtractorMOG2()#视频的背景分割,Foreground Segmentation Algorithm
size = (int(cap1.get(cv2.CAP_PROP_FRAME_WIDTH)), int(cap1.get(cv2.CAP_PROP_FRAME_HEIGHT)))
fourcc=cv2.VideoWriter_fourcc('X','V','I','D')
out = cv2.VideoWriter('output.avi',fourcc,20.0,size)
firstFrame1 = None
firstFrame2 = None
firstFrame3 = None
firstFrame4 = None
xdata=np.array((0,0,0,0))
ydata=np.array((0,0,0,0))
##俯视图划分成56*56的网格
#grid1=np.zeros((56,56))
#grid2=np.zeros((56,56))
#grid3=np.zeros((56,56))
#grid4=np.zeros((56,56))
#相机位置
c1=[0,0]
c2=[0,56]
c3=[56,0]
c4=[56,56]
count=0
while(1):
ret1, frame1 = cap1.read(0)
ret2, frame2 = cap2.read(0)
ret3, frame3 = cap3.read(0)
ret4, frame4 = cap4.read(0)
if ret1 == True:
fgmask1 = fgbg.apply(frame1)
fgmask1 = cv2.morphologyEx(fgmask1, cv2.MORPH_OPEN, kernel)
blured1 = cv2.bilateralFilter(fgmask1, 10, 5, 5)
blured2 = cv2.GaussianBlur(fgmask1, (3, 3), 0)
fgmask1 = blured2
fgmask1 = closing = cv2.morphologyEx(fgmask1, cv2.MORPH_CLOSE, kernel)
#寻找视频中的轮廓
im1, contours1, hierarchy1 = cv2.findContours(fgmask1, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
for c in contours1:
perimeter = cv2.arcLength(c,True)
if perimeter >250:
#找到一个直矩形(不会旋转)
x,y,w,h1 = cv2.boundingRect(c)
#画出这个矩形
cv2.rectangle(frame1,(x,y),(x+w,y+h1),(0,255,0),2)
fgmask2 = fgbg.apply(frame2)
fgmask2 = cv2.morphologyEx(fgmask2, cv2.MORPH_OPEN, kernel)
blured1 = cv2.bilateralFilter(fgmask2, 10, 5, 5)
blured2 = cv2.GaussianBlur(fgmask2, (3, 3), 0)
fgmask2 = blured2
fgmask2 = closing = cv2.morphologyEx(fgmask2, cv2.MORPH_CLOSE, kernel)
#寻找视频中的轮廓
im, contours2, hierarchy = cv2.findContours(fgmask2, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
for c in contours2:
perimeter = cv2.arcLength(c,True)
if perimeter >250:
#找到一个直矩形(不会旋转)
x,y,w,h2 = cv2.boundingRect(c)
#画出这个矩形
cv2.rectangle(frame2,(x,y),(x+w,y+h2),(0,255,0),2)
fgmask3 = fgbg.apply(frame3)
fgmask3 = cv2.morphologyEx(fgmask3, cv2.MORPH_OPEN, kernel)
blured1 = cv2.bilateralFilter(fgmask3, 10, 5, 5)
blured2 = cv2.GaussianBlur(fgmask3, (3, 3), 0)
fgmask3 = blured2
fgmask3 = closing = cv2.morphologyEx(fgmask3, cv2.MORPH_CLOSE, kernel)
#寻找视频中的轮廓
im, contours3, hierarchy = cv2.findContours(fgmask3, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
for c in contours3:
perimeter = cv2.arcLength(c,True)
if perimeter >250:
#找到一个直矩形(不会旋转)
x,y,w,h3 = cv2.boundingRect(c)
#画出这个矩形
cv2.rectangle(frame3,(x,y),(x+w,y+h3),(0,255,0),2)
fgmask4 = fgbg.apply(frame4)
fgmask4 = cv2.morphologyEx(fgmask4, cv2.MORPH_OPEN, kernel)
blured1 = cv2.bilateralFilter(fgmask4, 10, 5, 5)
blured2 = cv2.GaussianBlur(fgmask4, (3, 3), 0)
fgmask4 = blured2
fgmask4 = closing = cv2.morphologyEx(fgmask4, cv2.MORPH_CLOSE, kernel)
#寻找视频中的轮廓
im, contours4, hierarchy = cv2.findContours(fgmask4, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
for c in contours4:
perimeter = cv2.arcLength(c,True)
if perimeter >200:
#找到一个直矩形(不会旋转)
x,y,w,h4 = cv2.boundingRect(c)
#画出这个矩形
cv2.rectangle(frame4,(x,y),(x+w,y+h4),(0,255,0),2)
#单目测距
d1=0.01*h1/1.8
d2=0.01*h2/1.8
d3=0.01*h3/1.8
d4=0.01*h4/1.8
dmax=max(d1,d2,d3,d4)
d1=d1*56/dmax
d2=d2*56/dmax
d3=d3*56/dmax
d4=d4*56/dmax
#俯视图划分成56*56的网格
grid1=np.zeros((56,56))
grid2=np.zeros((56,56))
grid3=np.zeros((56,56))
grid4=np.zeros((56,56))
for i in range(56):
for j in range(56):
if (np.linalg.norm(np.array([i,j])-np.array(c1))<d1+1 and
np.linalg.norm(np.array([i,j])-np.array(c1))>d1-1):
grid1[i,j]=grid1[i,j]+np.linalg.norm(np.array([i,j])-np.array(c1))
if (np.linalg.norm(np.array([i,j])-np.array(c2))<d2+1 and
np.linalg.norm(np.array([i,j])-np.array(c2))>d2-1):
grid2[i,j]=grid2[i,j]+np.linalg.norm(np.array([i,j])-np.array(c2))
if (np.linalg.norm(np.array([i,j])-np.array(c3))<d3+1 and
np.linalg.norm(np.array([i,j])-np.array(c3))>d3-1):
grid3[i,j]=grid3[i,j]+np.linalg.norm(np.array([i,j])-np.array(c3))
if (np.linalg.norm(np.array([i,j])-np.array(c4))<d4+1 and
np.linalg.norm(np.array([i,j])-np.array(c4))>d4-1):
grid4[i,j]=grid4[i,j]+np.linalg.norm(np.array([i,j])-np.array(c4))
Grid1=grid1/np.max(grid1)
Grid2=grid2/np.max(grid2)
Grid3=grid3/np.max(grid3)
Grid4=grid4/np.max(grid4)
grid=np.array(grid1+grid2+grid3+grid4)
#找出概率最大值的位置ad1,ad2,ad3,ad4
ad1=[np.array(np.where(Grid1==np.max(Grid1)))[0,0],np.array(np.where(Grid1==np.max(Grid1)))[1,0]]
ad2=[np.array(np.where(Grid2==np.max(Grid2)))[0,0],np.array(np.where(Grid2==np.max(Grid2)))[1,0]]
ad3=[np.array(np.where(Grid3==np.max(Grid3)))[0,0],np.array(np.where(Grid3==np.max(Grid3)))[1,0]]
ad4=[np.array(np.where(Grid4==np.max(Grid4)))[0,0],np.array(np.where(Grid4==np.max(Grid4)))[1,0]]
#保存在向量xdata,ydata中
xdata=np.vstack((xdata,np.array((ad1[0],ad2[0],ad3[0],ad4[0]))))
ydata=np.vstack((ydata,np.array((ad1[1],ad2[1],ad3[1],ad4[1]))))
#out.wrte(fgmask1)
# cv2.namedWindow('frame1',cv2.WINDOW_NORMAL)#调整窗口
# cv2.imshow('frame1',frame1)
# cv2.namedWindow('fgmask1',cv2.WINDOW_NORMAL)#调整窗口
# cv2.imshow('fgmask1',fgmask1)
# cv2.namedWindow('frame2',cv2.WINDOW_NORMAL)#调整窗口
# cv2.imshow('frame2',frame2)
# cv2.namedWindow('fgmask2',cv2.WINDOW_NORMAL)#调整窗口
# cv2.imshow('fgmask2',fgmask2)
# cv2.namedWindow('frame3',cv2.WINDOW_NORMAL)#调整窗口
# cv2.imshow('frame3',frame3)
# cv2.namedWindow('fgmask3',cv2.WINDOW_NORMAL)#调整窗口
# cv2.imshow('fgmask3',fgmask3)
# cv2.namedWindow('frame4',cv2.WINDOW_NORMAL)#调整窗口
# cv2.imshow('frame4',frame4)
# cv2.namedWindow('fgmask4',cv2.WINDOW_NORMAL)#调整窗口
# cv2.imshow('fgmask4',fgmask4)
else:
break
if cv2.waitKey(25) & 0xFF == ord('q'):
break
out.release()
cap1.release()
cv2.destroyAllWindows()
#import numpy as np
#import matplotlib
#matplotlib.use("Agg")
##import matplotlib.pyplot as plt
#import matplotlib.animation as animation
#
##FFMpegWriter = manimation.writers['ffmpeg']
##metadata = dict(title='Movie Test', artist='Matplotlib',
## comment='Movie support!')
##writer = FFMpegWriter(fps=15, metadata=metadata)
#Writer = animation.writers['ffmpeg']
#writer = Writer(fps=15, metadata=dict(artist='Me'), bitrate=1800)
#
#fig = plt.figure()
#l, = plt.plot([], [], 'k-o')
#
#plt.xlim(-5, 5)
#plt.ylim(-5, 5)
#
#with writer.saving(fig, "writer_test.mp4", 100):
# for i in range(100):
# l.set_data(xdata[i,], ydata[i,])
# writer.grab_frame()
#fourcc=cv2.VideoWriter_fourcc('X','V','I','D')
#video = cv2.VideoWriter('top.avi',fourcc,5,size)
(n,p)=np.shape(xdata)
for i in range(n):
fig = plt.figure()
ax = fig.gca()
plt.axis([0, 56, 0, 56],'equal')
ax.set_xticks(np.arange(0, 56, 1))
ax.set_yticks(np.arange(0, 56, 1))
plt.scatter(xdata[i,],ydata[i,],c = ('r','b','g','k'),marker = 'o')
plt.grid()
fig.set_size_inches(18.5, 10.5)
img =plt.gcf()
#plt.show()
img.savefig('/Users/vicky/Desktop/第五问视频/img'+np.str(i)+'.jpg')
# img_data = np.asarray(img)
# video.write(img_data)
# img2 = np.empty([IMAGE_SIZE, IMAGE_SIZE, 3], dtype=type(RED[0]))
# cv2.imshow('video', img)
# ani = animation.FuncAnimation(img,update_img,300,interval=30)
# writer = animation.writers['ffmpeg'](fps=30)
# ani.save('demo.mp4',writer=writer,dpi=dpi)
#video.release()
#import glob as gb
#img_path = gb.glob("/Users/vicky/Desktop/第五问视频/img/.jpg")
##videoWriter = cv2.VideoWriter('test.mp4', cv2.VideoWriter_fourcc(*'MJPG'), 25, (640,480))
#videoWriter = cv2.VideoWriter('img.avi', fourcc, 20.0,size)
#
#for path in img_path:
# img = cv2.imread(path)
# #img = cv2.resize(img,(640,480))
# videoWriter.write(img)
#
#videoWriter.release()