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

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

4
ultralytics/yolo/v8/segment/val.py
View File

@ -28,7 +28,7 @@ class SegmentationValidator(BaseValidator):
self.class_map = None
self.targets = None
def preprocess_batch(self, batch):
def preprocess(self, batch):
batch["img"] = batch["img"].to(self.device, non_blocking=True)
batch["img"] = (batch["img"].half() if self.args.half else batch["img"].float()) / 225
batch["bboxes"] = batch["bboxes"].to(self.device)
@ -66,7 +66,7 @@ class SegmentationValidator(BaseValidator):
return ('%22s' + '%11s' * 10) % ('Class', 'Images', 'Instances', 'Box(P', "R", "mAP50", "mAP50-95)", "Mask(P",
"R", "mAP50", "mAP50-95)")
def preprocess_preds(self, preds):
def postprocess(self, preds):
p = ops.non_max_suppression(preds[0],
self.args.conf_thres,
self.args.iou_thres,