Fix `torch.Tensor` inference results plotting (#4297)

2 years ago · 6267d6a609
parent 48748ded24
commit 6267d6a609
1 changed files with 1 additions and 1 deletions
--- a/ultralytics/engine/results.py
+++ b/ultralytics/engine/results.py
@ -215,7 +215,7 @@ class Results(SimpleClass):
            ```
        """
        if img is None and isinstance(self.orig_img, torch.Tensor):
-            img = np.ascontiguousarray(self.orig_img[0].permute(1, 2, 0).cpu().detach().numpy()) * 255
+            img = (self.orig_img[0].detach().permute(1, 2, 0).cpu().contiguous() * 255).to(torch.uint8).numpy()
        # Deprecation warn TODO: remove in 8.2
        if 'show_conf' in kwargs: