我想在django中使用yolov5模型,但遇到了麻烦。
我想做的是,如果用户上传一张图片到django服务器,用yolov5模型进行对象检测,然后在web上显示结果。这个过程本身很简单。但是我不知道如何附加yolov5 api和django。
使用过yolo的人都知道,yolo基本上是基于命令运行api的。
!python train.py --img 512 --batch 16 --epochs 100 --data ~~ # for training
!python detect.py --weights'/content/yolov5/runs/~~ # for detection
有什么方法可以在Django的视图中运行这个吗?
我尝试了一些Python函数,比如execfile()和os.system(),它们在Python脚本中以解释器的方式执行命令,但是它们不能正常工作(我认为Django和Yolo之间的路径是扭曲的)。
事实上,如果可能的话,最好像Keras一样将yolo api作为模块加载,并像函数一样运行它,而不是命令方法。但是我找不到一种方法来像模块或类似的东西一样使用yolov5。
我怎样才能解决这个问题?请给予我一些建议。
1条答案
按热度按时间ars1skjm1#
使用Django和Yolov5进行对象检测
使用yolov5配置Django项目:
Models.py:
Views.py:
在www.example.com中有很多可以修改views.py,比如添加更多的函数和逻辑,但这里的目的是连接yolov5和Django。views.py中的配置是最重要的,因为它是yolov5 www.example.com文件的入口hubconf.py。
Forms.py
Imagemodel_form.html
简单的网页最初:
检测后: