yolov5-code-main/app.py

import torch
import gradio as gr

model = torch.hub.load("./", "custom", path="yolov5s.pt", source="local")

title = "✨基于YOLOV5的目标检测平台✨"

desc = "(●'◡'●)"
base_conf, base_iou = 0.15, 0.5


def det_image(img, conf_最小置信度, iou_最大置信度):
    model.conf = conf_最小置信度
    model.iou = iou_最大置信度

    return 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)],
             outputs=["image"],
             fn=det_image,
             title=title,
             description=desc,
             article = "(●ˇ∀ˇ●)",
             live=True,
             examples=[["E:/BBX/Document/pytorch/息肉病变检测/YOLO5/yolov5-master/data/images/301.jpg", base_conf, base_iou], ["E:/BBX/Document/pytorch/息肉病变检测/YOLO5/yolov5-master/data/images/402.jpg", 0.3, base_iou], ["E:/BBX/Document/pytorch/息肉病变检测/YOLO5/yolov5-master/data/images/503.jpg", base_conf, base_iou]]).launch(auth=("admin", "1234"),share=True)