xclele/yolov10
1
0
Fork 0
mirror of https://github.com/THU-MIG/yolov10.git synced 2025-05-23 21:44:22 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix segment coco val (#156)

Browse Source 
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
This commit is contained in:
Laughing 2023-01-08 20:51:10 +08:00 committed by GitHub
parent 07eab49c3d
commit c985eaba0d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -90,12 +90,14 @@ class SegmentationValidator(DetectionValidator):
if self.args.single_cls: if self.args.single_cls:
pred[:, 5] = 0 pred[:, 5] = 0
predn = pred.clone() predn = pred.clone()
ops.scale_boxes(batch["img"][si].shape[1:], predn[:, :4], shape) # native-space pred ops.scale_boxes(batch["img"][si].shape[1:], predn[:, :4], shape,
ratio_pad=batch["ratio_pad"][si]) # native-space pred
# Evaluate # Evaluate
if nl: if nl:
tbox = ops.xywh2xyxy(bbox) # target boxes tbox = ops.xywh2xyxy(bbox) # target boxes
ops.scale_boxes(batch["img"][si].shape[1:], tbox, shape) # native-space labels ops.scale_boxes(batch["img"][si].shape[1:], tbox, shape,
ratio_pad=batch["ratio_pad"][si]) # native-space labels
labelsn = torch.cat((cls, tbox), 1) # native-space labels labelsn = torch.cat((cls, tbox), 1) # native-space labels
correct_bboxes = self._process_batch(predn, labelsn) correct_bboxes = self._process_batch(predn, labelsn)
# TODO: maybe remove these `self.` arguments as they already are member variable # TODO: maybe remove these `self.` arguments as they already are member variable
@ -117,7 +119,9 @@ class SegmentationValidator(DetectionValidator):
# Save # Save
if self.args.save_json: if self.args.save_json:
pred_masks = ops.scale_image(batch["img"][si].shape[1:], pred_masks = ops.scale_image(batch["img"][si].shape[1:],
pred_masks.permute(1, 2, 0).contiguous().cpu().numpy(), shape) pred_masks.permute(1, 2, 0).contiguous().cpu().numpy(),
shape,
ratio_pad=batch["ratio_pad"][si])
self.pred_to_json(predn, batch["im_file"][si], pred_masks) self.pred_to_json(predn, batch["im_file"][si], pred_masks)
# if self.args.save_txt: # if self.args.save_txt:
# save_one_txt(predn, save_conf, shape, file=save_dir / 'labels' / f'{path.stem}.txt') # save_one_txt(predn, save_conf, shape, file=save_dir / 'labels' / f'{path.stem}.txt')