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

Integrating seq-nms with yolov5s model #7

Open
linsecDev opened this issue Apr 23, 2022 · 2 comments
Open

Integrating seq-nms with yolov5s model #7

linsecDev opened this issue Apr 23, 2022 · 2 comments

Comments

@linsecDev
Copy link

linsecDev commented Apr 23, 2022

Hi @tmoopenn , can you guide how and where to use this seq-nms repo on yolov5. I tried using the seq_nms function on ultralytics/yolov5 repo but function parameters don't correspond well.

@SanKumSan
Copy link

Hi @tmoopenn , could you also please provide the sample inputs for the boxes, scores as defined in the method.
boxes : Tensor of shape (num_frames, num_boxes, 4) containing the boxes in (x1, y1, x2, y2) format.
scores : Tensor of shape (num_frames, num_boxes) containing the confidence score for each box.

Thanks

@wch243294382
Copy link

have you finished this work?

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

3 participants