xdobro23/YOLOv8-16bit
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[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
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):