From fa3b1462de678901acd59adfb213a118590d2147 Mon Sep 17 00:00:00 2001 From: UltralyticsAssistant Date: Thu, 28 Nov 2024 22:12:19 +0000 Subject: [PATCH] Update Docs for 'ultralytics - 35e1944' --- reference/models/yolo/classify/predict/index.html | 2 +- reference/nn/modules/head/index.html | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/reference/models/yolo/classify/predict/index.html b/reference/models/yolo/classify/predict/index.html index 1b9cec285..e945d2c35 100644 --- a/reference/models/yolo/classify/predict/index.html +++ b/reference/models/yolo/classify/predict/index.html @@ -1243,7 +1243,7 @@

orig_imgs = ops.convert_torch2numpy_batch(orig_imgs) return [ - Results(orig_img, path=img_path, names=self.model.names, probs=pred) + Results(orig_img, path=img_path, names=self.model.names, probs=pred.softmax(0)) for pred, orig_img, img_path in zip(preds, orig_imgs, self.batch[0]) ] diff --git a/reference/nn/modules/head/index.html b/reference/nn/modules/head/index.html index 6533b7f04..d3b84cbcf 100644 --- a/reference/nn/modules/head/index.html +++ b/reference/nn/modules/head/index.html @@ -2212,7 +2212,7 @@

if isinstance(x, list): x = torch.cat(x, 1) x = self.linear(self.drop(self.pool(self.conv(x)).flatten(1))) - return x if self.training else x.softmax(1) + return x