編集履歴

質問編集履歴

ソースコード

2016/11/21 14:04

投稿

maru.

スコア7

test CHANGED Viewed

File without changes

test CHANGED Viewed

@@ -66,520 +66,6 @@
 ###該当のソースコード
-Python
-coding: UTF-8
-import cv2
-import cv2.cv as cv
-import numpy as np
-import os.path
-import sys
-import copy
-import csv
-import math
-fps = 30
-fourcc = cv2.cv.CV_FOURCC('m','p','4','v')
-delete_cord_x1 = 2770
-delete_cord_y1 = 1791
-delete_cord_x2 = 2970
-delete_cord_y2 = 1791
-font = cv2.FONT_ITALIC
-term_crit = ( cv2.TERM_CRITERIA_EPS | cv2.TERM_CRITERIA_COUNT, 10, 1 )
-cascade_path = "./Cascade/cascade.xml"
-cascade = cv2.CascadeClassifier(cascade_path)
-color_green = (0,255,0)
-color_blue = (255,0,0)
-color_red = (0,0,255)
-DetectInformations = []
-tmp_DetectInformations = []
-tmp = []
-argv = sys.argv
-argc = len(argv)
-if(argc != 2):
-		print 'Usage: python object_tracking.py FileName'
-		quit()
-def func():
-	f = open("new_od_data.csv","w")
-	csvWriter = csv.writer(f,lineterminator="\n")
-	firstline = ["ID","Origin","Destination","speed"]
-	csvWriter.writerow(firstline)
-	csv_array = []
-	cap = cv2.VideoCapture(argv[1])
-	width = int(cap.get(cv.CV_CAP_PROP_FRAME_WIDTH))
-	hight = int(cap.get(cv.CV_CAP_PROP_FRAME_HEIGHT))
-	goal_frame = int(cap.get(cv.CV_CAP_PROP_FRAME_COUNT))
-	tmp_name,ext = os.path.splitext(argv[1])
-	output_name = tmp_name+"new_tracked.m4v"
-	output = cv2.VideoWriter( output_name,fourcc,fps,(width,hight))
-	frame_num = 0
-	ID = 0
-	print 'goal is',goal_frame
-	while(cap.isOpened):
-		ret,frame = cap.read()
-		gray_frame = cv2.cvtColor(frame,cv2.cv.CV_BGR2GRAY)
-		if ret == False:
-			break
-		if len(DetectInformations) == 0:
-			rects = cascade.detectMultiScale(gray_frame, scaleFactor=1.11,minNeighbors=3,maxSize=(80,80),minSize=(30,30))
-			for rect in rects:
-				if rect[0] < delete_cord_x1 and rect[1] < delete_cord_y1:
-					pass
-				elif delete_cord_x2 < rect[0] and rect[1] < delete_cord_y2:
-					pass
-				else:
-					ID = ID + 1
-					DetectInformations.append(DetectInfo(rect,ID))
-			DrawRect(frame)
-		else:
-			rects = cascade.detectMultiScale(gray_frame, scaleFactor=1.11,minNeighbors=3,maxSize=(80,80),minSize=(30,30))
-			del tmp[:]
-			for rect in rects:
-				tmp.append(DetectInfo(rect,0))
-			for tmp_DI in tmp:
-				if tmp_DI.first_rect[0] < delete_cord_x1 and tmp_DI.first_rect[1] < delete_cord_y1:
-					tmp_DI.delete = True
-				if delete_cord_x2 < tmp_DI.first_rect[0] and tmp_DI.first_rect[1] < delete_cord_y2:
-					tmp_DI.delete = True
-			for DI in DetectInformations:
-				for tmp_DI in tmp:
-					if DI.now_rect[0] < delete_cord_x1 and DI.now_rect[1] < delete_cord_y1:
-						DI.delete = True
-					elif delete_cord_x2 < DI.now_rect[0] and DI.now_rect[1] < delete_cord_y2:
-						DI.delete = True
-					else:
-						distance = math.hypot(tmp_DI.first_rect[0] - DI.now_rect[0],tmp_DI.first_rect[1] - DI.now_rect[1])
-						if distance < 40:
-							tmp_DI.delete = True
-							DI.new = False
-							DI.now_rect = tmp_DI.first_rect
-						elif distance == 0:
-							DI.delete = True
-			for DI in DetectInformations:
-				movement = math.hypot(DI.first_rect[0] - DI.now_rect[0], DI.first_rect[1] - DI.now_rect[1])
-				if movement < 5:
-					DI.delete = True
-			for DI in DetectInformations:
-				if DI.delete:#DI.delete=True
-					Origin = CalCenter(DI.first_rect)
-					Destination = CalCenter(DI.now_rect)
-					speed = CalSpeed(Origin[0] ,Origin[1],Destination[0] ,Destination[1],DI.count)
-					line = [DI.ID,CalCenter(DI.first_rect),CalCenter(DI.now_rect),speed]
-					csvWriter.writerow(line)
-			DeleteTmp()
-			DeleteDetectInfo()
-		for tmp_DI in tmp:
-			ID = ID + 1
-			DetectInformations.append(DetectInfo(tmp_DI.first_rect,ID))
-		DrawRect(frame)
-		output.write(frame)
-		frame_num += 1
-		if(frame_num == goal_frame):
-			break
-		if (frame_num % 10 == 0):
-			print frame_num
-	###################動画処理終了###########################
-	text = "#########################################"
-	csvWriter.writerow(text)
-	for DI in DetectInformations:
-		Origin = CalCenter(DI.first_rect)
-		Destination = CalCenter(DI.now_rect)
-		speed = CalSpeed(Origin[0] ,Origin[1],Destination[0] ,Destination[1],DI.count)
-		line = [DI.ID,Origin,Destination,speed]
-		csvWriter.writerow(line)
-	cap.release()
-	output.release()
-	cv2.destroyAllWindows()
-	f.close()
-	print frame_num
-	print "output :",output_name,"od_data.csv"
-def DrawRect(frame):
-	for DI in DetectInformations:
-		DI.count = DI.count + 1
-		if DI.new == True:
-			cv2.rectangle(frame,tuple(DI.first_rect[0:2]),tuple(DI.first_rect[0:2]+DI.first_rect[2:4]),color_blue,1)
-			cv2.putText(frame,DI.ID,(DI.first_rect[0]+2,DI.first_rect[1]+2),font,0.65,color_red)
-			if DI.count > 5:
-				DI.delete = True
-				DeleteDetectInfo()
-		elif DI.new == False:
-			cv2.rectangle(frame,tuple(DI.now_rect[0:2]),tuple(DI.now_rect[0:2]+DI.now_rect[2:4]),color_green,1)
-			cv2.putText(frame,DI.ID,(DI.now_rect[0]+2,DI.now_rect[1]+2),font,0.65,color_red)
-def CalCenter(rect):
-	x = rect[0]+rect[2]/2
-	y = rect[1]+rect[3]/2
-	center = []
-	center.append(x)
-	center.append(y)
-	return center
-def CalSpeed(x1,y1,x2,y2,frame_count):
-	dist = math.hypot(x1 - x2,y1 - y2)
-	if frame_count == 0:
-		return 0
-	else:
-		speed = dist / frame_count
-		return speed
-def DeleteDetectInfo():
-	del tmp_DetectInformations[:]
-	for DI in DetectInformations:
-		if DI.delete == False:
-			tmp_DetectInformations.append(DI)
-	del DetectInformations[:]
-	for DI in tmp_DetectInformations:
-		DetectInformations.append(DI)
-def DeleteTmp():
-	del tmp_DetectInformations[:]
-	for DI in tmp:
-		if DI.delete == False:
-			tmp_DetectInformations.append(DI)
-	del tmp[:]
-	for DI in tmp_DetectInformations:
-		tmp.append(DI)
-class DetectInfo:
-	def __init__(self,rect,ID):
-		self.first_rect = rect
-		self.now_rect = rect
-		self.previous_rect = rect
-		self.ID = str(ID)
-		self.delete = False
-		self.new = True
-		self.count = 0
-		self.cmpcount = 0
-	def Get_now_rect(self):
-		return self.now_rect
-	def Get_first_rect(self):
-		return self.first_rect
-	def Get_ID(self):
-		return self.ID
-if __name__ == '__main__':
-	func()

2016/11/21 14:04

投稿

maru.

スコア7

test CHANGED Viewed

File without changes

test CHANGED Viewed

File without changes