YOLOv5在Django中调用

在不了解YOLOv5内部执行原理的前提下,想要在Django框架中调用YOLOv5,因为Django框架中的Views中允许调用自己编写的函数,利用OS库的os.popen构建cmd命令,调用Python终端进行执行。
再利用readlines获取输出的结果,其返回值为列表,为了方便获取想要的人数信息,先转为str再利用re正则表达式匹配人数前的数字。
我是利用yolo识别的图片中的人数,参数url是图片的文件路径,推荐使用绝对路径。想识别其他类别可以在cmd中修改–classes 后的参数,具体类别请参考YOLOv5官方文档
之后直接在views中调用自己封装好的函数即可。

def getperson(url:str) -> int:
	'''
	处理图片
	:param url:图片地址
	:return: 图片中的人数
	''
  • 2
    点赞
  • 22
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
Django是一个用于构建Web应用程序的高级Python Web框架,而Yolov5是一个流行的目标检测算法。想要在Django调用Yolov5,可以按照以下步骤进行: 1. 安装Yolov5:首先,需要按照Yolov5的要求安装它。可以通过使用pip包管理器执行以下命令来安装所需的依赖项: ``` pip install yolov5 ``` 2. 导入Yolov5模型:在Django项目的适当位置,将需要导入Yolov5模型的代码文件添加到项目。可以通过使用以下代码行导入模型: ``` import torch from models.experimental import attempt_load ``` 3. 加载和使用模型:在Django的视图函数或适当的位置,加载已训练的Yolov5模型并使用它来检测目标。可以使用以下代码行来加载模型和进行目标检测: ``` weights = 'path_to_yolov5_weights' # 替换为模型权重的路径 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu') model = attempt_load(weights, map_location=device) # 进行目标检测 results = model(source_image) # 替换为要检测的图像 ``` 4. 处理Yolov5检测结果:根据需要,可以使用Yolov5检测结果在Django执行特定的操作,例如可视化检测结果或将结果传递给前端界面。 需要注意的是,在整个过程,您需要确保正确配置Django项目以及正确安装和加载Yolov5模型。对于更复杂的应用程序,可能需要相应地调整代码和配置以满足您的需求。为了更好地处理图像上传和结果处理,您还可以考虑使用Django的Form和View类来处理图像上传和显示结果。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值