From 9d9c2c6af7f5b81b784bb570839d16da6ba77675 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Thu, 4 Aug 2022 20:13:02 +0200
Subject: [PATCH] Fix TensorRT --dynamic excess outputs bug

Potential fix for https://github.com/ultralytics/yolov5/issues/8790
---
 models/common.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/models/common.py b/models/common.py
index c898d94a921a..f93ef8736d7b 100644
--- a/models/common.py
+++ b/models/common.py
@@ -480,6 +480,8 @@ def forward(self, im, augment=False, visualize=False, val=False):
             self.binding_addrs['images'] = int(im.data_ptr())
             self.context.execute_v2(list(self.binding_addrs.values()))
             y = self.bindings['output'].data
+            if len(y) > b:
+                y = y[:b]  # trim excess --dynamic outputs
         elif self.coreml:  # CoreML
             im = im.permute(0, 2, 3, 1).cpu().numpy()  # torch BCHW to numpy BHWC shape(1,320,192,3)
             im = Image.fromarray((im[0] * 255).astype('uint8'))