Add YOLOv5 dataset yamls (#207)

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-01-10 18:33:44 +01:00
parent e371e81aa0
commit c7629e93bd
14 changed files with 2018 additions and 39 deletions
--- a/ultralytics/yolo/v8/classify/predict.py
+++ b/ultralytics/yolo/v8/classify/predict.py
@ -38,7 +38,7 @@ class ClassificationPredictor(BasePredictor):
        log_string += '%gx%g ' % im.shape[2:]  # print string
        self.annotator = self.get_annotator(im0)

-        prob = preds[idx]
+        prob = preds[idx].softmax(0)
        self.all_outputs.append(prob)
        # Print results
        top5i = prob.argsort(0, descending=True)[:5].tolist()  # top 5 indices