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

Can you point out the meaning of this number in label_anchor #18

Open
hieuth133 opened this issue Nov 4, 2021 · 2 comments
Open

Can you point out the meaning of this number in label_anchor #18

hieuth133 opened this issue Nov 4, 2021 · 2 comments

Comments

@hieuth133
Copy link

Thank you for the wonderful project. Can you please show what is the meaning of number 6 in this line of code?

https://github.com/LongxingTan/Yolov5/blob/88acfd988decc4cc78335cfb6eb50f1975294c1f/yolo/dataset/label_anchor.py#L35

@LongxingTan
Copy link
Owner

LongxingTan commented Nov 4, 2021

@hieuth133
The last dimension of y_true has 6 values.

  • the first four value is the box coordinates, like xywh or xyxy
  • the 5th value is the confidence of the detection box, for y_true it's 1
  • the 6th value is the classification label, in coco dataset it has 80 classes. In the loss function, this is expand into one-hot further

@hieuth133
Copy link
Author

Thank you!!

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