首页
编程
java
php
前端
首页
编程
java
php
前端
yolov7推理不出检测, detect.py, 训练自己的数据集,
yolov7训练自己的数据集,map能达到0.9,但是推理时(detect.py)不出检测框,之前用原来的yolov7.pt测试的时候是有框的
网上搜到的解决办法,half = False,cudnn。bechmark = True,都试了 也不行
推理时的环境是 win cudatoolkit 11.1 + cudnn 8.1 + torch1.8.0 + cu111
训练的环境是 ubuntu pytorch=1.11.0=py3.8_cuda11.3_cudnn8.2.0_0
点击展开全文