基于flask和网页端部署yolo自训练模型

Posted 2022-09-03 刘润森！

tags:

篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了基于flask和网页端部署yolo自训练模型相关的知识，希望对你有一定的参考价值。

@Author: Runsen

关于yolov5模型的flask的部署, 需要通过torch.hub.load 加载yolov5 自定义的模型，source需要设置local

if opt.model == "yolov5":
    model = torch.hub.load("yolo", "custom", path="models/yolov5s.pt", source=

以上是关于基于flask和网页端部署yolo自训练模型的主要内容，如果未能解决你的问题，请参考以下文章