python3+tensorflow实时读取rtmp的流+object_detection识别+ffmpeg推流直播

最新推荐文章于 2024-08-03 01:09:23 发布

weixin_42096284

最新推荐文章于 2024-08-03 01:09:23 发布

阅读量2.8k

点赞数

分类专栏：图像识别

本文链接：https://blog.csdn.net/weixin_42096284/article/details/102821560

版权

本文介绍了在Windows环境下，使用Python3和TensorFlow进行实时RTMP流物体检测，并通过FFmpeg推流直播的步骤。包括设置执行环境、下载并配置Nginx与FFmpeg，以及运行代码进行流处理。

摘要由CSDN通过智能技术生成

1.执行环境

win7(64) + GeForce940M 显卡+ cuda:10.0.130 + python3.7 + object_detection

2.下载Nginx

nginx 1.7.11.3 Gryphon下载连接
解压，在conf新增配置文件。复制nginx-win.conf文件改名为nginx-win-rtmp.conf，编辑nginx-win-rtmp.conf新增rtmp的配置

rtmp {
   
    server {
   
        listen 8080;
        chunk_size 4000;
        application live {
   
             live on;
 
             # record first 1K of stream
             record all;
             record_path /tmp/av;
             record_max_size 1K;
 
             # append current timestamp to each flv
             record_unique on;
 
             # publish only from localhost
             allow publish 127.0.0.1;
             deny publish all;
 
             #allow play all;
        }
    }
}

启动

nginx.exe -c conf\nginx-win-rtmp.conf

3.安装ffmpeg

下载地址为ffmpeg
配置好环境变量

4. 运行代码

import subprocess

import cv2
import numpy as np
import tensorflow as tf
from object_detection.utils import label_map_util
from object_detection.utils import visualization_utils as vis_util

PATH_TO_CKPT = r"E:\AI\frozen_inference_graph.pb"
PATH_TO_LABELS = r"E:\AI\mscoco_label_map.pbtxt"

NUM_CLASSES = 90

label_map = label_map_util.load_labelmap(PATH_TO_LABELS)
categories = label_map_util.convert_label_map_to_categories(
    label_map, max_num_classes=NUM_CLASSES, use_display_name=True)
category_index = label_map_util.create_category_index(categories)

detection_graph = tf.Graph