THU-MIG / yolov10

YOLOv10: Real-Time End-to-End Object Detection [NeurIPS 2024]
https://arxiv.org/abs/2405.14458
GNU Affero General Public License v3.0
9.94k stars 986 forks source link

能否获取框的分数? #189

Open BossCrab-jyj opened 5 months ago

BossCrab-jyj commented 5 months ago

class Results(SimpleClass): """ A class for storing and manipulating inference results.

Attributes:
    orig_img (numpy.ndarray): Original image as a numpy array.
    orig_shape (tuple): Original image shape in (height, width) format.
    boxes (Boxes, optional): Object containing detection bounding boxes.
    masks (Masks, optional): Object containing detection masks.
    probs (Probs, optional): Object containing class probabilities for classification tasks.
    keypoints (Keypoints, optional): Object containing detected keypoints for each object.
    speed (dict): Dictionary of preprocess, inference, and postprocess speeds (ms/image).
    names (dict): Dictionary of class names.
    path (str): Path to the image file.

这个probs是类别的分数,在yolov5里面分数等于框的分数*类别的分数,yolov10里面只有类别的分数吗? 请问能否得到框的分数?因为我对是否检测到目标更感兴趣。

jameslahm commented 5 months ago

感谢您的关注!您好,是的,yolov10里面只有类别的分数,您可以根据所有类别的分数的最大值是否大于阈值来判断是否检测到目标~

BossCrab-jyj commented 5 months ago

谢谢,但是我发现训练自己的数据时,小目标漏的会有很多,效果不如v5好,有什么好的方法和建议吗

jameslahm commented 5 months ago

您好,我们之前在这里 #136 对小目标检测提供了一些说明和建议,您方便看下这个是否有帮助吗?谢谢~