streamlit嵌入yoloV5

向streamlit移植yoloV5

本人原创，若要转载请附上连接

好久没有更新了，今天晚上更新我的第三篇😂

项目下载链接放于文末，有需要可自行下载

项目环境

pycharm2018
windowns10
python3.8
pytorch

项目组成

主要分成两部分：

1. yoloV5代码
2. streamlit构建简单的UI界面
3. 处理BUG

详细步骤

Yolo 方面

1、首先，我们需要前往 GitHub 下载所需的 YOLOV5 代码并按照readme.md 的使用方法进行环境部署
或者直接下载我的代码也可以运行(已下载s,l,x。3种权重文件)
2、 配置环境 cd 到 /yolo/下运行

pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple

3、 一定要先运行成功！
一定要先运行成功！
一定要先运行成功！
（重要的话说三遍）
4、运行过程中可能会出现包没有导入的错误，将所需的包导入即可
5、在下一篇博客中我会介绍一下yolo其中的参数设置以及含义

streamlit

1、streamlit我就不过多介绍，就是不需要HTML，CSS，Javascript也能将构建漂亮前端页面（对我这种怕麻烦的人就是一种福报）
2、使用
安装：

pip install streamlit

初步测试：

stream hello

出现如下图则表明你已经安装成功！

简单的运行.py文件方式

# 先进入py文件的目录下
streamlit run 你的文件名称.py

还有streamlit的API库介绍，官方API 供大家参考

二者整合

创建新的Project，并将Yolo文件以python package形式引入，并对Yolo的运行函数做封装，便于调用
根据自己的需要编写streamlit UI并添加事件

注意： BUG BUG BUG

Bug 1:下载我的代码包部署好后若发现导包处有错误，请不要改动，改动则运行出错。

例如：

Bug 2:若对视频进行检测时，发现检测后的视频无法用 streamlit.video(Path)进行播放，则改动 detect.py 200行左右

out = cv2.VideoWriter(save_path, cv2.VideoWriter_fourcc(*'avc3'), fps, (w, h))

如使用 ‘mp4v’ 进行编码则streamlit对视频不支持解码

Bug 3:indexFunction.py 文件中34，42行设置为自己的文件路径即可

。。。。可能还有其它地方，一时想不起，想到后我会补充在此

代码链接

Git: 代码下载
权重文件下载：https://cipfz.xyz/source/yolov5_weights.zip