Fetch the repository succeeded.
基于Streamlit的YOLOv5模型转换可视化工具
曾逸夫，从事人工智能研究与开发；主研领域：计算机视觉；YOLOv5官方开源项目代码贡献人；YOLOv5 v6.1代码贡献人；YOLOv5 v6.2代码贡献人；Gradio官方开源项目代码贡献人
❤️ Github：https://github.com/Zengyf-CVer
🔥 YOLOv5 官方开源项目PR ID：
--visualize：https://github.com/ultralytics/yolov5/pull/5701
detect.py --view-img for non-ASCII paths：https://github.com/ultralytics/yolov5/pull/7093
💡 YOLOv5 v6.1 & v6.2代码贡献链接：
🔥 Gradio 官方开源项目PR ID：
2022-08-14 ⚡ Streamlit YOLOv5 Model2X v0.3正式上线
2022-07-17 ⚡ Streamlit YOLOv5 Model2X v0.2正式上线
2022-07-15 ⚡ Streamlit YOLOv5 Model2X v0.1正式上线
本项目提供了在线demo，点击下面的logo，进入Hugging Face Spaces中快速体验：
❗ 注：tflite和tfjs需要分开导出；除CoreML外，其他half导出需要选中cuda:0
.
├── streamlit_yolov5_modle2x # 项目名称
│ ├── models # YOLOv5官方模型文件
│ ├── model_config # YOLOv5官方工具包
│ ├── val.py # YOLOv5官方验证文件
│ ├── export.py # YOLOv5官方导出文件
│ ├── __init__.py # 初始化文件
│ ├── app_01.py # 单选，单个模型转换和下载
│ ├── app_02.py # 多选，多项模型转换和打包下载
│ ├── app_03.py # 多选，多项模型转换和打包下载
│ ├── LICENSE # 项目许可
│ ├── CodeCheck.md # 代码检查
│ ├── .gitignore # git忽略文件
│ ├── setup.cfg # pre-commit CI检查源配置文件
│ ├── .pre-commit-config.yaml # pre-commit配置文件
│ ├── yolov5_pytorch_gpu.md # YOLOv5 PyTorch GPU安装教程
│ ├── README.md # 项目说明
│ └── requirements.txt # 脚本依赖包
conda create -n yolo python==3.8
conda activate yolo # 进入环境
git clone https://gitee.com/CV_Lab/streamlit_yolov5_modle2x.git
cd streamlit_yolov5_modle2x
pip install -r ./requirements.txt
# 注：requirements.txt中onnxruntime-gpu、tensorflow、nvidia-pyindex和nvidia-tensorrt均需要在GPU模式下安装
❗ 注意：yolov5默认采用pip安装PyTorch GPU版，如果采用官网安装PyTorch GPU版，参见YOLOv5 PyTorch GPU安装教程
streamlit run app_04.py # v0.3 推荐（适用于cuda:0）
streamlit run app_03.py # v0.2 推荐（适用于cpu）
streamlit run app_02.py # v0.1
streamlit run app_01.py # v0.1
| 编号 | 格式 | 参数 | 模型 | 用途 |
|---|---|---|---|---|
| 01 | PyTorch | - | yolov5s.pt |
- |
| 02 | TorchScript | torchscript |
yolov5s.torchscript |
移动端 |
| 03 | ONNX | onnx |
yolov5s.onnx |
通用 |
| 04 | OpenVINO | openvino |
yolov5s_openvino_model/ |
边缘 云 |
| 05 | TensorRT | engine |
yolov5s.engine |
边缘计算 |
| 06 | CoreML | coreml |
yolov5s.mlmodel |
Apple |
| 07 | TensorFlow SavedModel | saved_model |
yolov5s_saved_model/ |
完整的TF模型 |
| 08 | TensorFlow GraphDef | pb |
yolov5s.pb |
数据流图 |
| 09 | TensorFlow Lite | tflite |
yolov5s.tflite |
移动端 |
| 10 | TensorFlow Edge TPU | edgetpu |
yolov5s_edgetpu.tflite |
边缘计算 |
| 11 | TensorFlow.js | tfjs |
yolov5s_web_model/ |
web |
| 12 | PaddlePaddle | paddle |
yolov5s_paddle_model |
❗ 注：参考官方文档：https://github.com/ultralytics/yolov5/releases/tag/v6.1
❗ 注：在运行程序时可能会遇到的问题与解决办法
此处可能存在不合适展示的内容，页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容，可点击提交进行申诉，我们将尽快为您处理。
Activity
Community
Health
Trend
Influence
：Code submit frequency
：React/respond to issue & PR etc.
：Well-balanced team members and collaboration
：Recent popularity of project
：Star counts, download counts etc.