HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

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

Browse source
This commit is contained in:
Ayush Chaurasia 2022-11-09 22:05:15 +05:30 • committed by GitHub
parent 4c68b9dcf6
commit d143ac666f
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 7 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

2
ultralytics/yolo/v8/classify/val.py
View file

@ -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