在开发应用中,特别是视觉类应用中,很多时候需要输入本地视频或RTSP视频流,使用Python语言可以简单实现,注意这里的.mp4视频需要为H.264或H.265编码,如果不确定的话,建议用格式工厂转换一下编码。来来,直接看代码吧。这段代码来自社区例程中的头部姿势检测项目Python版本
# -*- coding: utf-8 -*-
import sys
import re
import cv2
from graph import *
import yolo3_resnet18_inference
import deepsort_inference
import client
import threading
import time
import datetime
import ChannelManager
# 获取命令输入中的视频地址或RTSP视频流地址
lenofUrl = len(sys.argv)
# The number of parameters is incorrect.
if lenofUrl <= 1:
print("[ERROR] Please input mp4/Rtsp URL")
sys.exit()
elif lenofUrl >= 3:
print("[ERROR] param input Error")
sys.exit()
# 正则匹配,找到.mp4视频或者RTSP视频流
URL = sys.argv[1]
# match Input parameter