detect.py about drenet HOT 9 OPEN

Why does detect.py have no results

你好学长，我在利用已有的权重检测新的图片时，利用detect.py监测出来的图片上没有标注结果，是什么原因？

from drenet.

哈喽@ymj11,

我刚测试了下，代码本身应该是没问题的。下面左图是测试图，右图是测试结果：

如果测试的时候没有检测结果，一种可能是检测结果置信度较低，可以适当减小--conf-thres的值；另一种可能是检测器本身就没能检测出船只小目标，这可能是因为测试图像与训练图像差异较大

from drenet.

学长 这个 代码需要什么修改的地方吗 ？
Namespace(weights='.\DRENet.pt', source='images/', img_size=512, conf_thres=0.25, iou_thres=0.45, device='', view_img=False, save_txt=False, save_conf=False, classes=None, agnostic_nms=False, augment=False, update=False, project='runs/detect', name='exp', exist_ok=False)

我从数据集中找了 200多张影像还是一张都检测不出来 ，程序是正常能跑通的

from drenet.

那有点奇怪，数据集是指Levir-ship数据集吗？你可以发我几张含船的图像我测下

from drenet.

学长您试一下

from drenet.

哈喽 @ymj11 ,

我这边输出结果如下（就是直接运行detect.py，没有作任何修改）：

from drenet.

好的 学长 我运行出来了 我把--device 参数写的cpu就可以运行出来 我直接用的gpu跑 程序也不会报错
YOLOv5 torch 2.1.0+cu121 CUDA:0 (NVIDIA GeForce GTX 1660 SUPER, 6143.6875MB)
这个原因怎么解释呀 学长

from drenet.

这我也不清楚，好奇怪。我这cpu和cuda都可以运行，按理说device设置不影响结果的。

from drenet.

好的 谢谢学长 麻烦您了 我在仔细研究一下

from drenet.