Add initial model interface (#30)

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-10-26 01:21:15 +05:30
parent 7b560f7861
commit 1054819a59
12 changed files with 220 additions and 109 deletions
--- a/ultralytics/yolo/v8/classify/val.py
+++ b/ultralytics/yolo/v8/classify/val.py
@ -1,9 +1,9 @@
 import torch

-from ultralytics import yolo
+from ultralytics.yolo.engine.validator import BaseValidator


-class ClassificationValidator(yolo.BaseValidator):
+class ClassificationValidator(BaseValidator):

    def init_metrics(self):
        self.correct = torch.tensor([])