yolov8cls 输出 - Githubissues

wang-xinyu / tensorrtx

Implementation of popular deep learning networks with TensorRT network definition API

MIT License

6.93k stars 1.77k forks source link

yolov8cls 输出 #1544

Open gopin95 opened 3 months ago

gopin95 commented 3 months ago

Env

GPU, e.g. V100, RTX2080, TX2, Xavier NX, Nano, etc.
OS, e.g. Ubuntu16.04, Win10, etc.
Cuda version
TensorRT version

About this repo

which branch/tag/commit are you using?
which model? yolov5, retinaface?

Your problem

yolov8cls输出是按照个数来输出的，我希望将输出由具体个数改成由置信度来决定输出目标个数，以保持和检测一致。

wang-xinyu commented 3 months ago

@lindsayshuo

stale[bot] commented 1 month ago

This issue has been automatically marked as stale because it has not had recent activity. It will be closed if no further activity occurs. Thank you for your contributions.