【YOLOV5 入门】——Gradio搭建Web GUI

2024-05-15 05:12
文章标签 入门 web 搭建 yolov5 gui gradio

本文主要是介绍【YOLOV5 入门】——Gradio搭建Web GUI,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

引入:上节搭建的UI可视化界面只能以运行程序弹出窗口的形式运行,不能在网页Web中使用,本次代码将会非常少!


一、Gradio简介与安装

Gradio 是一个用于构建机器学习模型演示界面和Web应用的开源库。提供了简单易用的界面,使您可以快速地将机器学习模型部署为交互式应用程序,而无需编写大量的前端代码。下面是一些关于 Gradio 的主要特点和功能:

  • 简单易用:Gradio 提供了简洁直观的 API,使得用户能够轻松地创建交互式界面。

  • 多样化的输入和输出:Gradio 支持多种类型的输入和输出,包括文本、图像、视频、音频和数据框。这使得您可以将各种类型的模型部署为交互式应用程序,例如图像分类、文本生成、对象检测等。

  • 自定义界面:Gradio 允许用户自定义界面的布局和样式,以满足特定的需求。您可以轻松地调整输入和输出部件的大小、排列方式和样式。

  • 实时更新:Gradio 提供了实时更新的功能,使得应用程序能够在用户与模型进行交互的同时实时更新输出结果。

  • 多模型集成:Gradio 支持将多个模型集成到同一个应用程序中,使用户能够比较不同模型的性能和结果。

  • 部署和共享:Gradio 允许用户将创建的应用程序部署到云端或本地服务器,并生成一个分享链接,方便用户与他人共享应用程序。

Pycharm终端进入虚拟环境,pip安装:

pip install gradio


二、gradio_demo搭建Web GUI

1、简单界面

import torch
import gradio as grmodel = torch.hub.load("./", "custom", "runs/train/exp9/weights/best", source="local")
gr.Interface(inputs=["image"], outputs=["image"], fn=lambda img:model(img).render()[0]).launch()
"""
用 Gradio 创建了一个界面,该界面接受一个图像作为输入,并输出一个图像:
gradio.Interface: 这是 Gradio 中用于创建交互式界面的主要类。您可以使用它来定义输入和输出以及应用程序的行为。
inputs=["image"]: 这表示输入部分只有一个元素,类型为图像。在这个界面中,用户将能够上传一张图像作为输入。实际用的inputs=[gr.Image]类,这里由于不需要变更直接使用字符串映射实例化
outputs=["image"]: 这表示输出部分也只有一个元素,同样是图像类型。这意味着用户将得到一张图像作为输出。
fn=lambda img:model(img).render()[0]: 这是定义了应用程序行为的部分。在这里,fn 参数接受一个函数,该函数将输入图像作为参数,然后返回一个图像作为输出。在这个函数中,model(img) 调用了一个模型,该模型接受图像作为输入并返回一个结果。.render()[0] 则表示从模型返回的结果中取出第一个元素,这可能是处理后的图像。
launch(): 这是用于启动 Gradio 应用程序的方法。一旦调用了 launch(),Gradio 将根据之前定义的输入、输出和行为创建一个界面,并在默认浏览器中打开该界面,以便用户与之交互。
"""

上面是一个最基本的代码demo,运行上面代码:

ctrl+左击链接可跳转web:


2、充实界面

现在界面太单一,下面完善代码:

import torch
import gradio as grmodel = torch.hub.load("./", "custom", "runs/train/exp9/weights/best", source="local")title = "基于Gradio搭建YOLOv5检测Web GUI"
desc = "这种方式搭建的UI界面代码量很少,不要太方便!!!"base_conf,base_iou = 0.35, 0.55  #定义一个基准的conf和iou参数,刚打开会设置为这个,点击web界面“clear”按钮也可返回默认值def det_image(img, conf_thresh, iou_thresh):model.conf = conf_thresh  #将滑动条对应的conf和iou参数给modelmodel.iou = iou_threshreturn model(img).render()[0]gr.Interface(inputs=["image",gr.Slider(minimum=0,maximum=1, value=base_conf),gr.Slider(minimum=0,maximum=1,value=base_iou)], #inputs=["image","slider", "slider"]slider为调整conf和iou参数的滑动条,默认时0-100,需要调整所以没用这种写法outputs=["image"],fn=det_image,title=title,description=desc).launch()

如果还想添加功能可继续完善下面函数,比如可以添加几张可供选择的示例图片:

.....
gr.Interface(inputs=["image",gr.Slider(minimum=0,maximum=1, value=base_conf),gr.Slider(minimum=0,maximum=1,value=base_iou)], #inputs=["image","slider", "slider"]slider为调整conf和iou参数的滑动条,默认时0-100,需要调整所以没用这种写法outputs=["image"],fn=det_image,title=title,description=desc,examples = [["./datasets/images/train/y.png", base_conf, base_iou],["./datasets/images/train/l.png", base_conf, base_iou]]).launch()

如果说想调用摄像头检测,可把input参数做调整:

....
gr.Interface(inputs=[gr.Webcam(),gr.Slider(minimum=0,maximum=1, value=base_conf),gr.Slider(minimum=0,maximum=1,value=base_iou)], #inputs=["image","slider", "slider"]slider为调整conf和iou参数的滑动条,默认时0-100,需要调整所以没用这种写法outputs=["image"],fn=det_image,title=title,description=desc,examples = [["./datasets/images/train/y.png", base_conf, base_iou],["./datasets/images/train/l.png", base_conf, base_iou]]).launch()

如果不想要每次点击submit提交,可以加一个参数live =True:

...
gr.Interface(inputs=["image",gr.Slider(minimum=0,maximum=1, value=base_conf),gr.Slider(minimum=0,maximum=1,value=base_iou)], #inputs=["image","slider", "slider"]slider为调整conf和iou参数的滑动条,默认时0-100,需要调整所以没用这种写法outputs=["image"],fn=det_image,title=title,description=desc,live= True,examples = [["./datasets/images/train/y.png", base_conf, base_iou],["./datasets/images/train/l.png", base_conf, base_iou]]).launch()

如果想将其生成一个地址,在公网的任何地方点击都可运行,可指定launch的参数share=True,这样其他人也可通过链接进行使用:

.....
gr.Interface(inputs=["image",gr.Slider(minimum=0,maximum=1, value=base_conf),gr.Slider(minimum=0,maximum=1,value=base_iou)], #inputs=["image","slider", "slider"]slider为调整conf和iou参数的滑动条,默认时0-100,需要调整所以没用这种写法outputs=["image"],fn=det_image,title=title,description=desc,live= True,examples = [["./datasets/images/train/y.png", base_conf, base_iou],["./datasets/images/train/l.png", base_conf, base_iou]]).launch(share=True)

由于服务器在国外,链接什么的加载较慢,可自行试验!

这种是最基础的Web GUI设计方法,如果有更多复杂需求可以使用其他工具!


往期精彩

STM32专栏(9.9)icon-default.png?t=N4N7http://t.csdnimg.cn/A3BJ2

OpenCV-Python专栏(9.9)icon-default.png?t=N4N7http://t.csdnimg.cn/jFJWe

AI底层逻辑专栏(9.9)icon-default.png?t=N4N7http://t.csdnimg.cn/6BVhM

机器学习专栏(免费)icon-default.png?t=N4N7http://t.csdnimg.cn/ALlLlSimulink专栏(免费)icon-default.png?t=N4N7http://t.csdnimg.cn/csDO4电机控制专栏(免费)icon-default.png?t=N4N7http://t.csdnimg.cn/FNWM7 

这篇关于【YOLOV5 入门】——Gradio搭建Web GUI的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/990903

相关文章

Spring Security 从入门到进阶系列教程

Spring Security 入门系列 《保护 Web 应用的安全》 《Spring-Security-入门(一):登录与退出》 《Spring-Security-入门(二):基于数据库验证》 《Spring-Security-入门(三):密码加密》 《Spring-Security-入门(四):自定义-Filter》 《Spring-Security-入门(五):在 Sprin

搭建Kafka+zookeeper集群调度

前言 硬件环境 172.18.0.5        kafkazk1        Kafka+zookeeper                Kafka Broker集群 172.18.0.6        kafkazk2        Kafka+zookeeper                Kafka Broker集群 172.18.0.7        kafkazk3

Java Web指的是什么

Java Web指的是使用Java技术进行Web开发的一种方式。Java在Web开发领域有着广泛的应用,主要通过Java EE(Enterprise Edition)平台来实现。  主要特点和技术包括: 1. Servlets和JSP:     Servlets 是Java编写的服务器端程序,用于处理客户端请求和生成动态网页内容。     JSP(JavaServer Pages)

数论入门整理(updating)

一、gcd lcm 基础中的基础,一般用来处理计算第一步什么的,分数化简之类。 LL gcd(LL a, LL b) { return b ? gcd(b, a % b) : a; } <pre name="code" class="cpp">LL lcm(LL a, LL b){LL c = gcd(a, b);return a / c * b;} 例题:

Java 创建图形用户界面(GUI)入门指南(Swing库 JFrame 类)概述

概述 基本概念 Java Swing 的架构 Java Swing 是一个为 Java 设计的 GUI 工具包,是 JAVA 基础类的一部分,基于 Java AWT 构建,提供了一系列轻量级、可定制的图形用户界面(GUI)组件。 与 AWT 相比,Swing 提供了许多比 AWT 更好的屏幕显示元素,更加灵活和可定制,具有更好的跨平台性能。 组件和容器 Java Swing 提供了许多

【IPV6从入门到起飞】5-1 IPV6+Home Assistant(搭建基本环境)

【IPV6从入门到起飞】5-1 IPV6+Home Assistant #搭建基本环境 1 背景2 docker下载 hass3 创建容器4 浏览器访问 hass5 手机APP远程访问hass6 更多玩法 1 背景 既然电脑可以IPV6入站,手机流量可以访问IPV6网络的服务,为什么不在电脑搭建Home Assistant(hass),来控制你的设备呢?@智能家居 @万物互联

BUUCTF靶场[web][极客大挑战 2019]Http、[HCTF 2018]admin

目录   [web][极客大挑战 2019]Http 考点:Referer协议、UA协议、X-Forwarded-For协议 [web][HCTF 2018]admin 考点:弱密码字典爆破 四种方法:   [web][极客大挑战 2019]Http 考点:Referer协议、UA协议、X-Forwarded-For协议 访问环境 老规矩,我们先查看源代码

poj 2104 and hdu 2665 划分树模板入门题

题意: 给一个数组n(1e5)个数,给一个范围(fr, to, k),求这个范围中第k大的数。 解析: 划分树入门。 bing神的模板。 坑爹的地方是把-l 看成了-1........ 一直re。 代码: poj 2104: #include <iostream>#include <cstdio>#include <cstdlib>#include <al

MySQL-CRUD入门1

文章目录 认识配置文件client节点mysql节点mysqld节点 数据的添加(Create)添加一行数据添加多行数据两种添加数据的效率对比 数据的查询(Retrieve)全列查询指定列查询查询中带有表达式关于字面量关于as重命名 临时表引入distinct去重order by 排序关于NULL 认识配置文件 在我们的MySQL服务安装好了之后, 会有一个配置文件, 也就

pico2 开发环境搭建-基于ubuntu

pico2 开发环境搭建-基于ubuntu 安装编译工具链下载sdk 和example编译example 安装编译工具链 sudo apt install cmake gcc-arm-none-eabi libnewlib-arm-none-eabi libstdc++-arm-none-eabi-newlib 注意cmake的版本,需要在3.17 以上 下载sdk 和ex