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

ultralytics 8.2.60 refactor process_mask_upsample (#14474)

Browse source 
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
Laughing 2024-07-19 01:25:38 +08:00 • committed by GitHub
parent 0822710185
commit dcde8bd23d
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
4 changed files with 2 additions and 27 deletions
Download patch file Download diff file Expand all files Collapse all files

2
ultralytics/models/yolo/segment/val.py
View file

@ -49,7 +49,7 @@ class SegmentationValidator(DetectionValidator):
if self.args.save_json:
check_requirements("pycocotools>=2.0.6")
# more accurate vs faster
self.process = ops.process_mask_upsample if self.args.save_json or self.args.save_txt else ops.process_mask
self.process = ops.process_mask_native if self.args.save_json or self.args.save_txt else ops.process_mask
self.stats = dict(tp_m=[], tp=[], conf=[], pred_cls=[], target_cls=[], target_img=[])
def get_desc(self):