增加多线程处理

.gitignore

@@ -0,0 +1,5 @@
+data/*
+snapshoot/*
+*.jpg
+*.pyc
+.vscode/*

AIServerDemo.py

@@ -0,0 +1,184 @@
+# -*- coding: UTF-8 -*-
+
+import numpy as np
+import cv2
+import sys
+import threading, _thread
+from queue import Queue
+import time
+import json
+import MTCNN_Crop_Face
+
+LIVE_URL = "rtsp://admin:admin123@172.16.1.29/cam/realmonitor?channel=1&subtype=0"
+RECORD_URL = "rtsp://admin:admin123@172.16.1.16:554/cam/playback?channel=1&subtype=0&starttime=2019_05_27_17_06_00&endtime=2019_05_27_17_08_00"
+
+SEGMENT_TIME = 60  #切片间隔，单位秒
+FPS = 15  #帧率，可以从码流中获取
+PER_FILE_FRAME = SEGMENT_TIME * FPS
+
+DATA_FILE_PATH = "data/"
+SNAP_FILE_PATH = "snapshoot/"
+
+MTC = MTCNN_Crop_Face.MTCNN_Crop_Face()
+
+class DealRecord():
+
+	def __init__(self, start_time, end_time):
+		# threading.Thread.__init__(self, name = "GetPicture")
+		self.start_time = start_time
+		self.end_time = end_time
+		start_timeArray = time.strptime(self.start_time, "%Y_%m_%d_%H_%M_%S")
+		sart_timeStamp = time.mktime(start_timeArray)
+		self.time_stamp = sart_timeStamp
+
+		self.face_cascade = cv2.CascadeClassifier('haarcascade_frontalface_default.xml')
+		self.frame_queue = Queue()
+
+		if '-' in self.end_time:
+			record_url = "rtsp://admin:admin123@172.16.1.16:554/cam/playback?channel=1&subtype=0&starttime=" + self.start_time
+		else:
+			record_url = "rtsp://admin:admin123@172.16.1.16:554/cam/playback?channel=1&subtype=0&starttime=" + self.start_time + "&endtime=" + self.end_time
+
+		self.capture = cv2.VideoCapture(record_url)
+		print("get %s"%(record_url))
+
+		self.stop_flag = False
+
+	def run(self):
+		# 打印视频相关参数，帧率，宽高
+		if self.capture.isOpened():
+			print (self.capture.get(cv2.CAP_PROP_FPS))
+			print (self.capture.get(cv2.CAP_PROP_FRAME_WIDTH))
+			print (self.capture.get(cv2.CAP_PROP_FRAME_HEIGHT))
+
+		_thread.start_new_thread(self.getRecord, ())
+
+		self.dealRecord()
+
+
+	def getRecord(self):
+		while self.capture.isOpened() and not self.stop_flag:
+			# Capture frame-by-frame
+			ret, frame = self.capture.read()
+			# cv2.imshow('image_ori', frame)
+			if frame is None:
+				self.frame_queue.put("fileover")
+				break
+			else:
+				# frame = cv2.resize(frame,(864,576))
+				self.frame_queue.put(frame)
+				# print("---- put frame.")
+
+		print("---- get record finish.")
+		self.capture.release()
+
+
+	def dealRecord(self):
+		frame_num = 0
+		data = []
+		frame_to_save = None
+		frame_to_save_num = 0
+		while 1:
+			# get接口默认为阻塞接口，会一直等待数据
+			frame = self.frame_queue.get()
+
+			if frame == "fileover":
+				break
+
+			frame_num += 1
+			# if (frame_num % FPS == 0) or (self.frame_queue.qsize() != 0):
+			if (frame_num % (FPS*10) == 0):
+				print("get frame %d. left %d"% (frame_num, self.frame_queue.qsize()))
+
+			# 人脸检测
+			# faces = self.MTC.cropface(frame)
+			# 隔帧检测
+			if frame_num % 2 == 1:
+				faces = MTC.cropface(frame)
+				last_face = faces
+			else:
+				faces = last_face
+
+			for bbox in faces:
+				cv2.rectangle(frame, (int(bbox[0]), int(bbox[1])), (int(bbox[2]), int(bbox[3])), (0, 0, 255), 2)
+
+			content = str(faces) + "\n"
+			if (len(faces) != 0):
+				# 快照间隔至少10秒
+				if (frame_num - frame_to_save_num > FPS*10) or (frame_to_save_num == 0):
+					frame_to_save = frame
+					frame_to_save_num = frame_num
+
+			data.append(content)
+
+			# 保存快照
+			if frame_to_save is not None:
+				self.savePicture(frame_to_save, frame_to_save_num)
+				frame_to_save = None
+
+			# 写入文件
+			if (len(data) == PER_FILE_FRAME):
+				self.saveData(data)
+				data = []
+
+			# cv2.imshow('image',frame)
+
+			if cv2.waitKey(1) & 0xFF == ord('q'):
+				break
+
+		self.stop_flag = True
+		if len(data) > 50:
+			self.saveData(data)
+
+		cv2.destroyAllWindows()
+		print("---- deal record finish. total frame %d"%(frame_num))
+
+
+	def saveData(self, data):
+		start_name = time.strftime("%Y%m%d_%H%M%S", time.localtime(self.time_stamp))
+		file_name = DATA_FILE_PATH + start_name + ".txt"
+		txt = open(file_name, "w")
+		for line in data:
+			txt.write(line)
+		txt.flush()
+		txt.close()
+		print("save to file %s"%(file_name))
+
+		self.time_stamp += SEGMENT_TIME
+
+
+	def savePicture(self, frame_to_save, frame_num):
+		#文件名精确到秒
+		start_name = time.strftime("%Y%m%d_%H%M", time.localtime(self.time_stamp))
+		sec = ((frame_num-1) % PER_FILE_FRAME) / FPS
+		print( "file_num=%d  sec=%d"%(frame_num, sec) )
+		file_name = "%s%s%02d.jpg"%(SNAP_FILE_PATH, start_name, sec)
+		cv2.imwrite(file_name, frame_to_save)
+		print("save to file %s"%(file_name))
+
+
+if __name__ == "__main__":
+
+	start_time = "2019_05_27_17_30_00"
+	end_time =  "2019_05_27_17_35_00"
+
+	if (len(sys.argv) == 2):
+		while 1:
+			timestamp = time.time()
+			start_time_str = time.strftime("%Y_%m_%d_%H_%M", time.localtime(timestamp - 60)) + "_00"
+			end_time_str = time.strftime("%Y_%m_%d_%H_%M", time.localtime(timestamp)) + "_00"
+			print("start=%s   end=%s"%(start_time_str, end_time_str))
+			test = DealRecord(start_time_str, end_time_str)
+			test.run()
+
+	if (len(sys.argv) == 3):
+		start_time = sys.argv[1]
+		end_time = sys.argv[2]
+
+	test = DealRecord(start_time, end_time)
+	test.run()
+
+	print("exit.")
+
+
+