fix: accelerate with gpu

RunCounting.py

@@ -14,6 +14,7 @@ from flask import Response
 from flask import Flask
 from flask import jsonify
 from flask_cors import CORS, cross_origin
+import subprocess
 
 t0 = time.time()
 
@@ -33,13 +34,13 @@ def counter():
 	global people_count
 	return jsonify([ 1000, 980, 700, 500, 670, 567, people_count ])
 
-@app.route("/video")
-@cross_origin()
-def video_feed():
-	# return the response generated along with the specific media
-	# type (mime type)
-	return Response(generate(),
-					mimetype = "multipart/x-mixed-replace; boundary=frame")
+# @app.route("/video")
+# @cross_origin()
+# def video_feed():
+# 	# return the response generated along with the specific media
+# 	# type (mime type)
+# 	return Response(generate(),
+# 					mimetype = "multipart/x-mixed-replace; boundary=frame")
 
 def open_ffmpeg_stream_process(height, width):
 	# args = (
@@ -48,7 +49,7 @@ def open_ffmpeg_stream_process(height, width):
 	#     "-f rtsp rtsp://172.23.90.205:8554/stream"
 	# ).split()
 	args = (
-			"ffmpeg -re -stream_loop -1 -f rawvideo -vcodec rawvideo -pix_fmt "
+			"ffmpeg -hwaccel cuda -re -stream_loop -1 -f rawvideo -vcodec rawvideo -pix_fmt "
 			"rgb24 -s " + str(width) + "x" + str(height) + " -i pipe:0 -pix_fmt nv12 "
 			"-c:v libx264 -preset fast -crf 22 -bf 0 "
 			"-f rtsp -rtsp_transport tcp rtsp://172.17.2.39/counter"
@@ -125,9 +126,7 @@ def PeopleCounter():
 
 	if config.Thread:
 		vs = thread.ThreadingClass(config.url)
-	ret, frame = vs.read()
-	height, width, ch = frame.shape
-	ffmpeg_process = open_ffmpeg_stream_process(height, width)
+	ffmpeg_process = open_ffmpeg_stream_process(1080, 1920)
 	# loop over frames from the video stream
 	while True:
 		# grab the next frame and handle if we are reading from either
@@ -143,7 +142,7 @@ def PeopleCounter():
 		# resize the frame to have a maximum width of 500 pixels (the
 		# less data we have, the faster we can process it), then convert
 		# the frame from BGR to RGB for dlib
-		frame = imutils.resize(frame, width = 500)
+		# frame = imutils.resize(frame, width = 640)
 		rgb = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
 
 		# if the frame dimensions are empty, set them
@@ -344,9 +343,10 @@ def PeopleCounter():
 			writer.write(frame)
 
 		# show the output frame
-
-		ffmpeg_process.stdin.write(frame.astype(np.uint8).tobytes())
 		# with lock:
+		frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
+		ffmpeg_process.stdin.write(frame.astype(np.uint8).tobytes())
+
 			# outputFrame = frame.copy()
 		# cv2.imshow("Real-Time Monitoring/Analysis Window", frame)
 		key = cv2.waitKey(1) & 0xFF
@@ -386,6 +386,8 @@ def PeopleCounter():
 		vs.release()
 
 	# close any open windows
+	ffmpeg_process.stdin.close()
+	ffmpeg_process.wait()
 	cv2.destroyAllWindows()
 
 def generate():