写在前面：当我们将模型训练出来了，此时就需要做UI界面给别人展示了。python提供的Gradio可以快速的搭建web页面。生成本地网址和公网网址，方面自己测试和用户测试。
目录
一、安装
二、搭建
三、代码详解
一、安装
Gradio
Gradio是一个开源的python库，用于构建机器学习演示和Web应用内置丰富的组件，并且实现了前后端的交互逻辑，无需额外编写代码环境安装
打开终端激活环境 conda activate yolov5(你的环境名)执行：pip install gradio
二、 搭建
1. 在yolov5-master目录下创建gradio_demo.py文件
执行代码
注意：
path:修改成你自己训练的模型examples中属性你可以删除了，也可以将里面预设的图片地址换成你自己的ps : 如果你有耐心的话，文末我会对这个代码进行解释!
import torchimport gradio as grmodel = torch.hub.load("./", "custom", path="mydata/weights/birds/best.pt", source="local")title = "鸟类视频检测Yolov5模型测试"desc = "这是一个基于Gradio的Yolov5演示项目，非常方便!"base_conf, base_iou = 0.25, 0.45def det_image(img, conf_thres, iou_thres): model.conf = conf_thres model.iou = iou_thres return model(img).render()[0]# examples中的参数要和inputs中对应# 获取摄像头拍照检测，修改inputs中的：inputs=[gr.Webcam(),...]就可以了，动态的更新添加属性：live=True# 如果将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)], outputs=["image"], fn=det_image, title=title, description=desc, live=True, examples=[["./mydata/test/image/Black_Footed_Albatross_0001_796111.jpg", base_conf, base_iou], ["./mydata/test/image/Gadwall_0001_31235.jpg", 0.3, base_iou]]).launch(share=True)
效果展示
三、代码详解
model = torch.hub.load("./", "custom", path="mydata/weights/birds/best.pt", source="local") 这段代码使用了PyTorch的torch.hub.load()函数来加载一个自定义模型。让我们逐步解释每个参数的作用：
./：这是模型所在的本地目录的路径。在这种情况下，./表示当前工作目录。如果你的模型文件在当前工作目录中，可以使用相对路径或绝对路径指定模型文件的位置。
"custom"：这是要加载的模型的名称或URL。它指定了你要加载的自定义模型的标识符。具体的标识符取决于模型的来源和用法，可以是模型的名称、URL或其他适当的标识符。
path="mydata/weights/birds/best.pt"：这是自定义模型文件的路径。它指定了模型文件的位置，相对于当前工作目录。
source="local"：这个参数指定了模型的来源。在这种情况下，source="local"表示模型文件位于本地文件系统中。你也可以使用其他来源，例如URL或远程资源。
def det_image(img, conf_thres, iou_thres): model.conf = conf_thres model.iou = iou_thres return model(img).render()[0]这段代码定义了一个名为 det_image 的函数，该函数接受三个参数：img、conf_thres 和 iou_thres。让我们逐个解释函数的每一行代码：
model.conf = conf_thres：这行代码将 model 对象的 conf 属性设置为 conf_thres 变量的值。这里假设 model 是一个对象或模型实例，并且它有一个名为 conf 的属性用于存储置信度阈值。
model.iou = iou_thres：这行代码将 model 对象的 iou 属性设置为 iou_thres 变量的值。同样，这里假设 model 是一个对象或模型实例，并且它有一个名为 iou 的属性用于存储 IoU（交并比）阈值。
return model(img).render()[0]：这行代码调用了 model 对象的 __call__() 方法，并将 img 作为参数传递给该方法。这意味着 model 对象被当作可调用对象，类似于函数的使用方式。然后，调用结果被传递给 render() 方法，并获取返回的结果。假设这个结果是一个可迭代对象，通过 [0] 取得其中的第一个元素，并将其作为函数的返回值。
这里的参数名：conf_thres,iou_thres是自定义的，但是页面上也有所关联。会和下图的名字对应。
title = "鸟类视频检测Yolov5模型测试"desc = "这是一个基于Gradio的Yolov5演示项目，非常方便!"base_conf, base_iou = 0.25, 0.45# examples中的参数要和inputs中对应# 获取摄像头拍照检测，修改inputs中的：inputs=[gr.Webcam(),...]就可以了，动态的更新添加属性：live=True# 如果将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)], outputs=["image"], fn=det_image, title=title, description=desc, live=True, examples=[["./mydata/test/image/Black_Footed_Albatross_0001_796111.jpg", base_conf, base_iou], ["./mydata/test/image/Gadwall_0001_31235.jpg", 0.3, base_iou]]).launch(share=True) 这段代码使用了gr.Interface()函数创建了一个基于Web的图形用户界面（Graphical Interface）来展示和使用det_image函数的结果。具体参数作用：
inputs=["image", gr.Slider(minimum=0, maximum=1, value=base_conf), gr.Slider(minimum=0, maximum=1, value=base_iou)]：这是一个输入项的列表，用于定义用户界面中的输入组件。在这个例子中，有三个输入项：一个名为 "image" 的图像输入项，以及两个滑动条输入项，分别用于设置 base_conf 和 base_iou 的值。滑动条的取值范围为 0 到 1。
outputs=["image"]：这是一个输出项的列表，用于定义用户界面中的输出组件。在这个例子中，只有一个输出项，即 "image"，用于显示处理后的图像结果。
fn=det_image：这是一个函数，用于处理输入项的值并生成输出项。在这个例子中，det_image 函数被指定为处理函数，它将接收输入项的值并生成相应的图像输出。
title=title：这是用户界面的标题，它的值通过变量 title 来指定。
description=desc：这是用户界面的描述，它的值通过变量 desc 来指定。
live=True：这个参数指定用户界面是实时的，即当输入项的值发生变化时，会立即进行处理和更新输出项。
examples=[["./mydata/test/image/Black_Footed_Albatross_0001_796111.jpg", base_conf, base_iou], ["./mydata/test/image/Gadwall_0001_31235.jpg", 0.3, base_iou]]：这是一个示例列表，用于提供一些预定义的输入值作为用户界面的初始值和示例。每个示例包含一个图像路径、一个置信度阈值和一个 IoU 阈值。
.launch(share=True)：这个方法用于启动用户界面，并将 share=True 设置为启用共享模式，这样其他人可以通过共享的链接来访问和使用该界面。
PS：yolov5避坑专栏，也有一些其他的文章，或许对你也有些帮助！让我们变的更强！！