search

Huangdebo / Confluence

A Robust Non-IoU Alternative to Non-Maxima Suppression in Object Detection
50 stars 10 forks source link

在yolov5上效果 #1

Open wuzuiyuzui opened 3 years ago

wuzuiyuzui commented 3 years ago

你好，我把您的代码移植到yolov5上发现在VOC数据集上并没有提升，而且还有了下降。请问你有做过相关实验吗

Huangdebo commented 3 years ago

你好，我把您的代码移植到yolov5上发现在VOC数据集上并没有提升，而且还有了下降。请问你有做过相关实验吗

可以上下调整一下那个 p_thres，默认是 0.6，试过在一些数据集中改成 0.7 会更好一些

Derrick9702 commented 3 years ago

你好，我把您的代码移植到yolov5上发现在VOC数据集上并没有提升，而且还有了下降。请问你有做过相关实验吗

可以上下调整一下那个 p_thres，默认是 0.6，试过在一些数据集中改成 0.7 会更好一些

您好，想请问一下您有尝试过移植在模型训练时候的test部分吗？还是说照常用nms进行训练，最后用测试集推理的时候才用confluence进行推理？我将您的代码移植到YOLOv5之后，在训练一个epoch之后进行test，要跑10多分钟才能输出一张图片的验证结果。

Huangdebo commented 3 years ago

你好，我把您的代码移植到yolov5上发现在VOC数据集上并没有提升，而且还有了下降。请问你有做过相关实验吗

可以上下调整一下那个 p_thres，默认是 0.6，试过在一些数据集中改成 0.7 会更好一些

您好，想请问一下您有尝试过移植在模型训练时候的test部分吗？还是说照常用nms进行训练，最后用测试集推理的时候才用confluence进行推理？我将您的代码移植到YOLOv5之后，在训练一个epoch之后进行test，要跑10多分钟才能输出一张图片的验证结果。

YOLO 的训练中是没有用到 nms 的，只有测试的时候会用到。confluence 的时间复杂度是 O(n*(n-1))，比 nms 复杂挺多的。你在test 时把 conf_thres 的值设置高一些，可以明显提速的。

duanyp12 commented 2 years ago

你好，我把您的代码移植到yolov5上发现在VOC数据集上并没有提升，而且还有了下降。请问你有做过相关实验吗

能提供下移植后的代码吗？我移植到yolov5上,验证时精度一直为0.

htcccc0096 commented 12 months ago

你好，把您的代码移植到yolov5上，运行val.py的时候出错，AttributeError: 'NoneType' object has no attribute 'shape'

Huangdebo commented 11 months ago

你好，把您的代码移植到yolov5上，运行val.py的时候出错，AttributeError: 'NoneType' object has no attribute 'shape'

应该是调用 shape 的那个变量没有赋值好，所以就会是 NoneType，建议先检查一下那个变量的值

htcccc0096 commented 11 months ago

你好，请问可以用于v8吗？需要修改代码吗？

Huangdebo commented 10 months ago

你好，请问可以用于v8吗？需要修改代码吗？

应该可以的，参考 confluence_process 这个函数，把传给 confluence 的 prediction 转化成相应的数据格式就可以了。