当前位置: 首页 > news >正文

利用flask将yolov5算法封装成在线推理服务

本脚本主要参考了yolov5工程文件夹下面的detect.py,将yolov5算法封装成了一个在线的推理服务,可以接受app请求,然后推理图片,并将检测结果以json返回,该服务可以供数据标注平台请求。


from flask import *
import shutil
import json
import os
import pynvml
import pandas as pd
import glob
import time
import cv2
import base64
import subprocess
import socket#下面的是yolov5需要import的库。直接从detect.py复制过来的。
import argparse
import os
import sys
from pathlib import Pathimport cv2
import torch
import torch.backends.cudnn as cudnnFILE = Path(__file__).resolve()
ROOT = FILE.parents[0]  # YOLOv5 root directory
if str(ROOT) not in sys.path:sys.path.append(str(ROOT))  # add ROOT to PATH
ROOT = Path(os.path.relpath(ROOT, Path.cwd()))  # relativefrom models.common import DetectMultiBackend
from utils.datasets import IMG_FORMATS, VID_FORMATS, LoadImages, LoadStreams
from utils.general import (LOGGER, check_file, check_img_size, check_imshow, check_requirements, colorstr,
http://www.lryc.cn/news/253591.html

相关文章:

  • vue3父子传值实现弹框功能
  • C++入门【2-C++ 数据类型】
  • 按照官网文档 通过useExtendedLib扩展库 引入WeUI,报错 组件未定义 | 解决办法
  • Chat-GPT原理
  • GODOC命令无效,原因是需要手动安装
  • 忽略python运行出现的大量警告
  • 【Polar靶场WEB签到】
  • Linux详解——常用命令(二)
  • TCP首部格式_基本知识
  • MIT线性代数笔记-第23讲-微分方程,exp(At)
  • windows下安装配置kafka
  • TV遥控器模拟鼠标键
  • 检测判断IP合法性API接口
  • Linux swatch命令教程:如何监控系统活动(附案例详解和注意事项)
  • 加州大学伯克利分校研究人员推出Starling-7B:一款通过人工智能反馈强化学习(RLAIF)训练的开源大型语言模型(LLM)
  • 腾讯面试真题(C语言)
  • JavaScript 函数
  • 数据结构 | 查漏补缺之DFS、BFS、二次探测再散列法、完全二叉树、深度计算
  • 用python实现单链表的基础操作
  • [头歌系统数据库实验] 实验3 MySQL的DDL语言
  • 系统运维安全之病毒自检及防护
  • Mabatis处理异常屏蔽SQL返回前端全局异常捕获处理
  • 黑豹程序员-java发邮件,发送内容支持html,带多附件的案例
  • [LeetCode] 15. 三数之和
  • Android Chips(标签)
  • 飞行汽车开发原理(上)
  • 22、pytest多个参数化的组合
  • 【网络奇缘】- 如何自己动手做一个五类|以太网|RJ45|网络电缆
  • 【从零开始学习JVM | 第三篇】类的生命周期(高频面试)
  • 详解前后端交互时PO,DTO,VO模型类的应用场景