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
-
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