ultralytics 8.0.153 YOLO Tasks Cleanup (#4314)

2023-08-12 02:30:57 +02:00 · 2023-08-12 02:30:57 +02:00 · 822608986c
commit 822608986c
parent 39395aedc8
22 changed files with 87 additions and 55 deletions
--- a/ultralytics/models/yolo/classify/val.py
+++ b/ultralytics/models/yolo/classify/val.py
@ -14,6 +14,8 @@ class ClassificationValidator(BaseValidator):
    def __init__(self, dataloader=None, save_dir=None, pbar=None, args=None, _callbacks=None):
        """Initializes ClassificationValidator instance with args, dataloader, save_dir, and progress bar."""
        super().__init__(dataloader, save_dir, pbar, args, _callbacks)
+        self.targets = None
+        self.pred = None
        self.args.task = 'classify'
        self.metrics = ClassifyMetrics()