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

8
ultralytics/yolo/engine/validator.py
View file

@ -54,7 +54,7 @@ class BaseValidator:
self.batch_i = batch_i
# pre-process
with dt[0]:
batch = self.preprocess_batch(batch)
batch = self.preprocess(batch)
# inference
with dt[1]:
@ -69,7 +69,7 @@ class BaseValidator:
# pre-process predictions
with dt[3]:
preds = self.preprocess_preds(preds)
preds = self.postprocess(preds)
self.update_metrics(preds, batch)
@ -89,10 +89,10 @@ class BaseValidator:
return stats
def preprocess_batch(self, batch):
def preprocess(self, batch):
return batch
def preprocess_preds(self, preds):
def postprocess(self, preds):
return preds
def init_metrics(self):