[rename] - preprocess-batch -> preprocess, preprocess_preds -> postprocess (#42)

2022-11-09 22:05:15 +05:30
parent 4c68b9dcf6
commit d143ac666f
3 changed files with 7 additions and 7 deletions
--- a/ultralytics/yolo/v8/classify/val.py
+++ b/ultralytics/yolo/v8/classify/val.py
@ -8,7 +8,7 @@ class ClassificationValidator(BaseValidator):
    def init_metrics(self, model):
        self.correct = torch.tensor([])

-    def preprocess_batch(self, batch):
+    def preprocess(self, batch):
        batch["img"] = batch["img"].to(self.device)
        batch["cls"] = batch["cls"].to(self.device)
        return batch