Skip to content

Commit

Permalink
Merge pull request #1818 from usamaimdadsian/tensorrt-batch-features-…
Browse files Browse the repository at this point in the history
…issue

Fix: Mistake in handling the output features of ReID model in TensorRT
  • Loading branch information
mikel-brostrom authored Feb 14, 2025
2 parents 0ef36ef + fc9b510 commit 0404fa0
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion boxmot/appearance/backends/tensorrt_backend.py
Original file line number Diff line number Diff line change
Expand Up @@ -116,7 +116,7 @@ def forward(self, im_batch):
# Execute inference
self.context.execute_v2(list(self.binding_addrs.values()))
features = self.bindings["output"].data
resultant_features.append(features)
resultant_features.append(features.clone())

if len(resultant_features)== 1:
return resultant_features[0]
Expand Down

0 comments on commit 0404fa0

Please sign in to comment.