You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
54 lines
1.2 KiB
54 lines
1.2 KiB
4 years ago
|
from AIDetector_pytorch import Detector
|
||
|
|
import imutils
|
||
|
|
import cv2
|
||
|
|
|
||
|
|
def main():
|
||
|
|
|
||
|
|
func_status = {}
|
||
|
|
func_status['headpose'] = None
|
||
|
|
|
||
|
|
name = 'demo'
|
||
|
|
|
||
|
|
det = Detector()
|
||
|
|
cap = cv2.VideoCapture('E:/视频/行人监控/test01.mp4')
|
||
|
|
fps = int(cap.get(5))
|
||
|
|
print('fps:', fps)
|
||
|
|
t = int(1000/fps)
|
||
|
|
|
||
|
|
size = None
|
||
|
|
videoWriter = None
|
||
|
|
|
||
|
|
while True:
|
||
|
|
|
||
|
|
# try:
|
||
|
|
_, im = cap.read()
|
||
|
|
if im is None:
|
||
|
|
break
|
||
|
|
|
||
|
|
result = det.feedCap(im, func_status)
|
||
|
|
result = result['frame']
|
||
|
|
result = imutils.resize(result, height=500)
|
||
|
|
if videoWriter is None:
|
||
|
|
fourcc = cv2.VideoWriter_fourcc(
|
||
|
|
'm', 'p', '4', 'v') # opencv3.0
|
||
|
|
videoWriter = cv2.VideoWriter(
|
||
|
|
'result.mp4', fourcc, fps, (result.shape[1], result.shape[0]))
|
||
|
|
|
||
|
|
videoWriter.write(result)
|
||
|
|
cv2.imshow(name, result)
|
||
|
|
cv2.waitKey(t)
|
||
|
|
|
||
|
|
if cv2.getWindowProperty(name, cv2.WND_PROP_AUTOSIZE) < 1:
|
||
|
|
# 点x退出
|
||
|
|
break
|
||
|
|
# except Exception as e:
|
||
|
|
# print(e)
|
||
|
|
# break
|
||
|
|
|
||
|
|
cap.release()
|
||
|
|
videoWriter.release()
|
||
|
|
cv2.destroyAllWindows()
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
|
||
|
|
main()
|