Replies: 1 comment
-
|
@bdv29 0, 1 and 2 in your screenshot are part of 3. See TensorRT Deployment Tutorial for exporting to TensorRT: YOLOv5 Tutorials
|
Beta Was this translation helpful? Give feedback.
0 replies
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Uh oh!
There was an error while loading. Please reload this page.
-
So I converted a custom trained model to TensorRT to run on the Jetson nano,
and the output I get is a list with four arrays,
the fourth array seems to be the one that gives me the correct bounding boxes when reshaped correctly taking account the amount of classes I have trained it for and fed into the NMS function ,
but I'm wondering is what are the other three arrays for? Is it just some odd idiosyncrasy of tensorRT or is there another aspect of yolo that I don't understand and could those other arrays have useful information?
screenshot of the four arrays

Beta Was this translation helpful? Give feedback.
All reactions