Yolov5 算子
Yolov5
对象检测算子 在检测到对象的图像上生成边界框。
更多信息在此: https://github.com/ultralytics/yolov5
Yolov5
尚未在模拟中进行微调,而是直接从 Pytorch Hub 导入权重信息。
如果您想在没有互联网的情况下运行yolov5
,你可以 克隆 https://github.com/ultralytics/yolov5 并 从 releases 页面 下载您要使用的权重文件,然后在 yaml 图中指定两个环境变量:
YOLOV5_PATH: 您的本地Yolov5路径
YOLOV5_WEIGHT_PATH: 您的本地 Yolov5 权重文件路径
您还可使用环境变量在GPU中选择分配模型:
PYTORCH_DEVICE: cuda # 或 cpu
输入
- 图像: 高 x 宽 x BGR array.
输出
- bbox: N_BBOX, X_MIN, X_MAX, Y_MIN, Y_MAX, CONDIDENCE, CLASS, array