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