Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

能否获取框的分数? #189

Open
BossCrab-jyj opened this issue Jun 4, 2024 · 3 comments
Open

能否获取框的分数? #189

BossCrab-jyj opened this issue Jun 4, 2024 · 3 comments

Comments

@BossCrab-jyj
Copy link

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
Copy link
Collaborator

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

@BossCrab-jyj
Copy link
Author

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

@jameslahm
Copy link
Collaborator

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

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants