使用opencv打开streamlit中st.file_uploader上传的视频

uBeetles

已于 2022-06-30 08:38:29 修改

阅读量2.6k

点赞数 7

文章标签： opencv 音视频计算机视觉

于 2022-06-29 18:12:20 首次发布

本文链接：https://blog.csdn.net/qq_45926473/article/details/125527086

版权

该博客介绍了一种解决在Streamlit应用中使用file_uploader上传视频后，因文件格式转换导致无法用opencv处理的问题。通过创建临时文件，将BytesIO格式的视频转换回opencv可读取的格式，实现了视频帧的逐帧处理和展示。代码示例展示了如何利用tempfile和cv2.VideoCapture读取并显示视频帧。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

由于在streanlit中使用st.file_uploader上传视频后，视频文件变为BytesIO的格式无法使用opencv打开并逐帧进行处理。
我们使用创造临时文件的方法来实现opencv处理上传文件，代码如下

import streamlit as st
import tempfile
import cv2

image_placeholder = st.empty()  # 创建空白块使得图片展示在同一位置

f = st.file_uploader("Upload file")  # 上传本地视频
if f is not None:
    tfile = tempfile.NamedTemporaryFile(delete=False)
    tfile.write(f.read())

    cap = cv2.VideoCapture(tfile.name)  # opencv打开文件

    if (cap.isOpened() == False):
        st.write("Error opening video stream or file")

    while (cap.isOpened()):
        success, frame = cap.read()
        if success:
            to_show = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
            image_placeholder.image(to_show, caption='Video')  # 将图片帧展示在同一位置得到视频效果
        else:
            break
    cap.release()