pytorch - 在自定义检测应用程序中使用 yolo V5 权重的最简单方法是什么？
问题描述
我已经在自定义数据集上训练了一个 yolov5 模型，现在我需要在我自己的应用程序中使用权重来检测对象，我只需要一个函数来接收训练后的模型权重和图像，并输出分类和边界框，我尝试修改 yolov5 GitHub 中提供的 detect.py 脚本，发现它使用了许多帮助函数，我尝试将所有内容集成到一个脚本中，但一直遇到导入错误。有没有更简单的方法来做到这一点？
解决方案
推荐阅读
- javascript - 检查 X 数组的所有值是否都在 Y 中
- laravel - 查询中的错误数据，Laravel
- react-native - 状态更改后反应原生不重新渲染
- python - 使用 Pandas 规范化 JSON 将路径附加到列
- python - 带有日期时间的 Python os.rename 不断保存多个同名文件
- c++ - 给定数字的其他 2 个毕达哥拉斯三元组
- c# - 首先创建代码 postgres 数据库失败：未提供密码，但后端需要密码（在 MD5 中）
- php - Apache、PHP 5.6 或 7.0：未加载库：openssl/lib/libcrypto.1.0.0.dylib 使用 Homebrew
- django - 基于国外模型字段值django的唯一约束
- google-chrome-devtools - DevTools - 关闭问题警告？