ultralytics 8.0.72__py3-none-any.whl → 8.0.74__py3-none-any.whl

ultralytics/__init__.py

@@ -1,6 +1,6 @@
 # Ultralytics YOLO 🚀, GPL-3.0 license
 
-__version__ = '8.0.72'
+__version__ = '8.0.74'
 
 from ultralytics.hub import start
 from ultralytics.yolo.engine.model import YOLO

ultralytics/hub/session.py

@@ -112,19 +112,21 @@ class HUBTrainingSession:
                 raise ValueError('Dataset may still be processing. Please wait a minute and try again.')  # RF fix
             self.model_id = data['id']
 
-            # TODO: restore when server keys when dataset URL and GPU train is working
-
-            self.train_args = {
-                'batch': data['batch_size'],
-                'epochs': data['epochs'],
-                'imgsz': data['imgsz'],
-                'patience': data['patience'],
-                'device': data['device'],
-                'cache': data['cache'],
-                'data': data['data']}
-
-            self.model_file = data.get('cfg', data['weights'])
-            self.model_file = checks.check_yolov5u_filename(self.model_file, verbose=False)  # YOLOv5->YOLOv5u
+            if data['status'] == 'new':  # new model to start training
+                self.train_args = {
+                    # TODO: deprecate 'batch_size' key for 'batch' in 3Q23
+                    'batch': data['batch' if ('batch' in data) else 'batch_size'],
+                    'epochs': data['epochs'],
+                    'imgsz': data['imgsz'],
+                    'patience': data['patience'],
+                    'device': data['device'],
+                    'cache': data['cache'],
+                    'data': data['data']}
+                self.model_file = data.get('cfg', data['weights'])
+                self.model_file = checks.check_yolov5u_filename(self.model_file, verbose=False)  # YOLOv5->YOLOv5u
+            elif data['status'] == 'training':  # existing model to resume training
+                self.train_args = {'data': data['data'], 'resume': True}
+                self.model_file = data['resume']
 
             return data
         except requests.exceptions.ConnectionError as e:

ultralytics/nn/modules.py

@@ -374,7 +374,7 @@ class Ensemble(nn.ModuleList):
         y = [module(x, augment, profile, visualize)[0] for module in self]
         # y = torch.stack(y).max(0)[0]  # max ensemble
         # y = torch.stack(y).mean(0)  # mean ensemble
-        y = torch.cat(y, 1)  # nms ensemble
+        y = torch.cat(y, 2)  # nms ensemble, y shape(B, HW, C)
         return y, None  # inference, train output
 
 

ultralytics/nn/tasks.py

@@ -539,7 +539,7 @@ def guess_model_task(model):
         model (nn.Module) or (dict): PyTorch model or model configuration in YAML format.
 
     Returns:
-        str: Task of the model ('detect', 'segment', 'classify').
+        str: Task of the model ('detect', 'segment', 'classify', 'pose').
 
     Raises:
         SyntaxError: If the task of the model could not be determined.

ultralytics/yolo/data/augment.py

@@ -127,7 +127,7 @@ class Mosaic(BaseMixTransform):
         s = self.imgsz
         yc, xc = (int(random.uniform(-x, 2 * s + x)) for x in self.border)  # mosaic center x, y
         for i in range(4):
-            labels_patch = (labels if i == 0 else labels['mix_labels'][i - 1]).copy()
+            labels_patch = labels if i == 0 else labels['mix_labels'][i - 1]
             # Load image
             img = labels_patch['img']
             h, w = labels_patch.pop('resized_shape')
@@ -223,18 +223,18 @@ class RandomPerspective:
 
     def affine_transform(self, img, border):
         # Center
-        C = np.eye(3)
+        C = np.eye(3, dtype=np.float32)
 
         C[0, 2] = -img.shape[1] / 2  # x translation (pixels)
         C[1, 2] = -img.shape[0] / 2  # y translation (pixels)
 
         # Perspective
-        P = np.eye(3)
+        P = np.eye(3, dtype=np.float32)
         P[2, 0] = random.uniform(-self.perspective, self.perspective)  # x perspective (about y)
         P[2, 1] = random.uniform(-self.perspective, self.perspective)  # y perspective (about x)
 
         # Rotation and Scale
-        R = np.eye(3)
+        R = np.eye(3, dtype=np.float32)
         a = random.uniform(-self.degrees, self.degrees)
         # a += random.choice([-180, -90, 0, 90])  # add 90deg rotations to small rotations
         s = random.uniform(1 - self.scale, 1 + self.scale)
@@ -242,12 +242,12 @@ class RandomPerspective:
         R[:2] = cv2.getRotationMatrix2D(angle=a, center=(0, 0), scale=s)
 
         # Shear
-        S = np.eye(3)
+        S = np.eye(3, dtype=np.float32)
         S[0, 1] = math.tan(random.uniform(-self.shear, self.shear) * math.pi / 180)  # x shear (deg)
         S[1, 0] = math.tan(random.uniform(-self.shear, self.shear) * math.pi / 180)  # y shear (deg)
 
         # Translation
-        T = np.eye(3)
+        T = np.eye(3, dtype=np.float32)
         T[0, 2] = random.uniform(0.5 - self.translate, 0.5 + self.translate) * self.size[0]  # x translation (pixels)
         T[1, 2] = random.uniform(0.5 - self.translate, 0.5 + self.translate) * self.size[1]  # y translation (pixels)
 
@@ -274,7 +274,7 @@ class RandomPerspective:
         if n == 0:
             return bboxes
 
-        xy = np.ones((n * 4, 3))
+        xy = np.ones((n * 4, 3), dtype=bboxes.dtype)
         xy[:, :2] = bboxes[:, [0, 1, 2, 3, 0, 3, 2, 1]].reshape(n * 4, 2)  # x1y1, x2y2, x1y2, x2y1
         xy = xy @ M.T  # transform
         xy = (xy[:, :2] / xy[:, 2:3] if self.perspective else xy[:, :2]).reshape(n, 8)  # perspective rescale or affine
@@ -282,7 +282,7 @@ class RandomPerspective:
         # create new boxes
         x = xy[:, [0, 2, 4, 6]]
         y = xy[:, [1, 3, 5, 7]]
-        return np.concatenate((x.min(1), y.min(1), x.max(1), y.max(1))).reshape(4, n).T
+        return np.concatenate((x.min(1), y.min(1), x.max(1), y.max(1)), dtype=bboxes.dtype).reshape(4, n).T
 
     def apply_segments(self, segments, M):
         """apply affine to segments and generate new bboxes from segments.
@@ -298,7 +298,7 @@ class RandomPerspective:
         if n == 0:
             return [], segments
 
-        xy = np.ones((n * num, 3))
+        xy = np.ones((n * num, 3), dtype=segments.dtype)
         segments = segments.reshape(-1, 2)
         xy[:, :2] = segments
         xy = xy @ M.T  # transform
@@ -319,7 +319,7 @@ class RandomPerspective:
         n, nkpt = keypoints.shape[:2]
         if n == 0:
             return keypoints
-        xy = np.ones((n * nkpt, 3))
+        xy = np.ones((n * nkpt, 3), dtype=keypoints.dtype)
         visible = keypoints[..., 2].reshape(n * nkpt, 1)
         xy[:, :2] = keypoints[..., :2].reshape(n * nkpt, 2)
         xy = xy @ M.T  # transform

ultralytics/yolo/data/base.py

@@ -3,6 +3,7 @@
 import glob
 import math
 import os
+from copy import deepcopy
 from multiprocessing.pool import ThreadPool
 from pathlib import Path
 from typing import Optional
@@ -177,13 +178,11 @@ class BaseDataset(Dataset):
         return self.transforms(self.get_label_info(index))
 
     def get_label_info(self, index):
-        label = self.labels[index].copy()
+        label = deepcopy(self.labels[index])  # requires deepcopy() https://github.com/ultralytics/ultralytics/pull/1948
         label.pop('shape', None)  # shape is for rect, remove it
         label['img'], label['ori_shape'], label['resized_shape'] = self.load_image(index)
-        label['ratio_pad'] = (
-            label['resized_shape'][0] / label['ori_shape'][0],
-            label['resized_shape'][1] / label['ori_shape'][1],
-        )  # for evaluation
+        label['ratio_pad'] = (label['resized_shape'][0] / label['ori_shape'][0],
+                              label['resized_shape'][1] / label['ori_shape'][1])  # for evaluation
         if self.rect:
             label['rect_shape'] = self.batch_shapes[self.batch[index]]
         label = self.update_labels_info(label)

ultralytics/yolo/data/utils.py

@@ -17,7 +17,8 @@ from PIL import ExifTags, Image, ImageOps
 from tqdm import tqdm
 
 from ultralytics.nn.autobackend import check_class_names
-from ultralytics.yolo.utils import DATASETS_DIR, LOGGER, NUM_THREADS, ROOT, clean_url, colorstr, emojis, yaml_load
+from ultralytics.yolo.utils import (DATASETS_DIR, LOGGER, NUM_THREADS, ROOT, SETTINGS_YAML, clean_url, colorstr, emojis,
+                                    yaml_load)
 from ultralytics.yolo.utils.checks import check_file, check_font, is_ascii
 from ultralytics.yolo.utils.downloads import download, safe_download, unzip_file
 from ultralytics.yolo.utils.ops import segments2boxes
@@ -246,6 +247,7 @@ def check_det_dataset(dataset, autodownload=True):
             if s and autodownload:
                 LOGGER.warning(m)
             else:
+                m += f"\nNote dataset download directory is '{DATASETS_DIR}'. You can update this in '{SETTINGS_YAML}'"
                 raise FileNotFoundError(m)
             t = time.time()
             if s.startswith('http') and s.endswith('.zip'):  # URL

ultralytics/yolo/engine/model.py

@@ -166,7 +166,9 @@ class YOLO:
         """
         Raises TypeError is model is not a PyTorch model
         """
-        if not isinstance(self.model, nn.Module):
+        pt_str = isinstance(self.model, (str, Path)) and Path(self.model).suffix == '.pt'
+        pt_module = isinstance(self.model, nn.Module)
+        if not (pt_module or pt_str):
             raise TypeError(f"model='{self.model}' must be a *.pt PyTorch model, but is a different type. "
                             f'PyTorch models can be used to train, val, predict and export, i.e. '
                             f"'yolo export model=yolov8n.pt', but exported formats like ONNX, TensorRT etc. only "
@@ -356,7 +358,6 @@ class YOLO:
             raise AttributeError("Dataset required but missing, i.e. pass 'data=coco128.yaml'")
         if overrides.get('resume'):
             overrides['resume'] = self.ckpt_path
-
         self.task = overrides.get('task') or self.task
         self.trainer = TASK_MAP[self.task][1](overrides=overrides, _callbacks=self.callbacks)
         if not overrides.get('resume'):  # manually set model only if not resuming

ultralytics/yolo/engine/predictor.py

@@ -109,8 +109,35 @@ class BasePredictor:
     def preprocess(self, img):
         pass
 
-    def write_results(self, results, batch, print_string):
-        raise NotImplementedError('print_results function needs to be implemented')
+    def write_results(self, idx, results, batch):
+        p, im, _ = batch
+        log_string = ''
+        if len(im.shape) == 3:
+            im = im[None]  # expand for batch dim
+        self.seen += 1
+        if self.source_type.webcam or self.source_type.from_img:  # batch_size >= 1
+            log_string += f'{idx}: '
+            frame = self.dataset.count
+        else:
+            frame = getattr(self.dataset, 'frame', 0)
+        self.data_path = p
+        self.txt_path = str(self.save_dir / 'labels' / p.stem) + ('' if self.dataset.mode == 'image' else f'_{frame}')
+        log_string += '%gx%g ' % im.shape[2:]  # print string
+        result = results[idx]
+        log_string += result.verbose()
+
+        if self.args.save or self.args.show:  # Add bbox to image
+            plot_args = dict(line_width=self.args.line_thickness, boxes=self.args.boxes)
+            if not self.args.retina_masks:
+                plot_args['im_gpu'] = im[idx]
+            self.plotted_img = result.plot(**plot_args)
+        # write
+        if self.args.save_txt:
+            result.save_txt(f'{self.txt_path}.txt', save_conf=self.args.save_conf)
+        if self.args.save_crop:
+            result.save_crop(save_dir=self.save_dir / 'crops', file_name=self.data_path.stem)
+
+        return log_string
 
     def postprocess(self, preds, img, orig_img):
         return preds

ultralytics/yolo/engine/results.py

@@ -7,13 +7,14 @@ Usage: See https://docs.ultralytics.com/modes/predict/
 
 from copy import deepcopy
 from functools import lru_cache
+from pathlib import Path
 
 import numpy as np
 import torch
 
 from ultralytics.yolo.data.augment import LetterBox
 from ultralytics.yolo.utils import LOGGER, SimpleClass, deprecation_warn, ops
-from ultralytics.yolo.utils.plotting import Annotator, colors
+from ultralytics.yolo.utils.plotting import Annotator, colors, save_one_box
 
 
 class BaseTensor(SimpleClass):
@@ -233,6 +234,80 @@ class Results(SimpleClass):
 
         return annotator.result()
 
+    def verbose(self):
+        """
+        Return log string for each tasks.
+        """
+        log_string = ''
+        probs = self.probs
+        boxes = self.boxes
+        if len(self) == 0:
+            return log_string if probs is not None else log_string + '(no detections), '
+        if probs is not None:
+            n5 = min(len(self.names), 5)
+            top5i = probs.argsort(0, descending=True)[:n5].tolist()  # top 5 indices
+            log_string += f"{', '.join(f'{self.names[j]} {probs[j]:.2f}' for j in top5i)}, "
+        if boxes:
+            for c in boxes.cls.unique():
+                n = (boxes.cls == c).sum()  # detections per class
+                log_string += f"{n} {self.names[int(c)]}{'s' * (n > 1)}, "
+        return log_string
+
+    def save_txt(self, txt_file, save_conf=False):
+        """Save predictions into txt file.
+
+        Args:
+            txt_file (str): txt file path.
+            save_conf (bool): save confidence score or not.
+        """
+        boxes = self.boxes
+        masks = self.masks
+        probs = self.probs
+        kpts = self.keypoints
+        texts = []
+        if probs is not None:
+            # classify
+            n5 = min(len(self.names), 5)
+            top5i = probs.argsort(0, descending=True)[:n5].tolist()  # top 5 indices
+            [texts.append(f'{probs[j]:.2f} {self.names[j]}') for j in top5i]
+        elif boxes:
+            # detect/segment/pose
+            for j, d in enumerate(boxes):
+                c, conf, id = int(d.cls), float(d.conf), None if d.id is None else int(d.id.item())
+                line = (c, *d.xywhn.view(-1))
+                if masks:
+                    seg = masks[j].xyn[0].copy().reshape(-1)  # reversed mask.xyn, (n,2) to (n*2)
+                    line = (c, *seg)
+                if kpts is not None:
+                    kpt = (kpts[j][:, :2] / d.orig_shape[[1, 0]]).reshape(-1).tolist()
+                    line += (*kpt, )
+                line += (conf, ) * save_conf + (() if id is None else (id, ))
+                texts.append(('%g ' * len(line)).rstrip() % line)
+
+        with open(txt_file, 'a') as f:
+            for text in texts:
+                f.write(text + '\n')
+
+    def save_crop(self, save_dir, file_name=Path('im.jpg')):
+        """Save cropped predictions to `save_dir/cls/file_name.jpg`.
+
+        Args:
+            save_dir (str | pathlib.Path): Save path.
+            file_name (str | pathlib.Path): File name.
+        """
+        if self.probs is not None:
+            LOGGER.warning('Warning: Classify task do not support `save_crop`.')
+            return
+        if isinstance(save_dir, str):
+            save_dir = Path(save_dir)
+        if isinstance(file_name, str):
+            file_name = Path(file_name)
+        for d in self.boxes:
+            save_one_box(d.xyxy,
+                         self.orig_img.copy(),
+                         file=save_dir / self.names[int(d.cls)] / f'{file_name.stem}.jpg',
+                         BGR=True)
+
 
 class Boxes(BaseTensor):
     """
@@ -339,6 +414,8 @@ class Masks(BaseTensor):
     """
 
     def __init__(self, masks, orig_shape) -> None:
+        if masks.ndim == 2:
+            masks = masks[None, :]
         self.masks = masks  # N, h, w
         self.orig_shape = orig_shape
 

ultralytics/yolo/engine/trainer.py

@@ -552,7 +552,7 @@ class BaseTrainer:
         if self.resume:
             assert start_epoch > 0, \
                 f'{self.args.model} training to {self.epochs} epochs is finished, nothing to resume.\n' \
-                f"Start a new training without --resume, i.e. 'yolo task=... mode=train model={self.args.model}'"
+                f"Start a new training without resuming, i.e. 'yolo train model={self.args.model}'"
             LOGGER.info(
                 f'Resuming training from {self.args.model} from epoch {start_epoch + 1} to {self.epochs} total epochs')
         if self.epochs < start_epoch:

ultralytics/yolo/utils/__init__.py

@@ -490,6 +490,7 @@ def get_user_config_dir(sub_dir='Ultralytics'):
 
 
 USER_CONFIG_DIR = Path(os.getenv('YOLO_CONFIG_DIR', get_user_config_dir()))  # Ultralytics settings dir
+SETTINGS_YAML = USER_CONFIG_DIR / 'settings.yaml'
 
 
 def emojis(string=''):
@@ -591,7 +592,7 @@ def set_sentry():
             logging.getLogger(logger).setLevel(logging.CRITICAL)
 
 
-def get_settings(file=USER_CONFIG_DIR / 'settings.yaml', version='0.0.3'):
+def get_settings(file=SETTINGS_YAML, version='0.0.3'):
     """
     Loads a global Ultralytics settings YAML file or creates one with default values if it does not exist.
 
@@ -640,7 +641,7 @@ def get_settings(file=USER_CONFIG_DIR / 'settings.yaml', version='0.0.3'):
         return settings
 
 
-def set_settings(kwargs, file=USER_CONFIG_DIR / 'settings.yaml'):
+def set_settings(kwargs, file=SETTINGS_YAML):
     """
     Function that runs on a first-time ultralytics package installation to set up global settings and create necessary
     directories.

ultralytics/yolo/utils/callbacks/hub.py

@@ -40,7 +40,7 @@ def on_model_save(trainer):
         # Upload checkpoints with rate limiting
         is_best = trainer.best_fitness == trainer.fitness
         if time() - session.timers['ckpt'] > session.rate_limits['ckpt']:
-            LOGGER.info(f'{PREFIX}Uploading checkpoint {session.model_id}')
+            LOGGER.info(f'{PREFIX}Uploading checkpoint https://hub.ultralytics.com/models/{session.model_id}')
             session.upload_model(trainer.epoch, trainer.last, is_best)
             session.timers['ckpt'] = time()  # reset timer
 

ultralytics/yolo/utils/ops.py

@@ -81,7 +81,8 @@ def segment2box(segment, width=640, height=640):
     x, y = segment.T  # segment xy
     inside = (x >= 0) & (y >= 0) & (x <= width) & (y <= height)
     x, y, = x[inside], y[inside]
-    return np.array([x.min(), y.min(), x.max(), y.max()]) if any(x) else np.zeros(4)  # xyxy
+    return np.array([x.min(), y.min(), x.max(), y.max()], dtype=segment.dtype) if any(x) else np.zeros(
+        4, dtype=segment.dtype)  # xyxy
 
 
 def scale_boxes(img1_shape, boxes, img0_shape, ratio_pad=None):
@@ -529,7 +530,8 @@ def resample_segments(segments, n=1000):
         s = np.concatenate((s, s[0:1, :]), axis=0)
         x = np.linspace(0, len(s) - 1, n)
         xp = np.arange(len(s))
-        segments[i] = np.concatenate([np.interp(x, xp, s[:, i]) for i in range(2)]).reshape(2, -1).T  # segment xy
+        segments[i] = np.concatenate([np.interp(x, xp, s[:, i]) for i in range(2)],
+                                     dtype=np.float32).reshape(2, -1).T  # segment xy
     return segments
 
 

ultralytics/yolo/v8/classify/predict.py

@@ -5,14 +5,10 @@ import torch
 from ultralytics.yolo.engine.predictor import BasePredictor
 from ultralytics.yolo.engine.results import Results
 from ultralytics.yolo.utils import DEFAULT_CFG, ROOT
-from ultralytics.yolo.utils.plotting import Annotator
 
 
 class ClassificationPredictor(BasePredictor):
 
-    def get_annotator(self, img):
-        return Annotator(img, example=str(self.model.names), pil=True)
-
     def preprocess(self, img):
         img = (img if isinstance(img, torch.Tensor) else torch.from_numpy(img)).to(self.model.device)
         return img.half() if self.model.fp16 else img.float()  # uint8 to fp16/32
@@ -27,43 +23,6 @@ class ClassificationPredictor(BasePredictor):
 
         return results
 
-    def write_results(self, idx, results, batch):
-        p, im, im0 = batch
-        log_string = ''
-        if len(im.shape) == 3:
-            im = im[None]  # expand for batch dim
-        self.seen += 1
-        im0 = im0.copy()
-        if self.source_type.webcam or self.source_type.from_img:  # batch_size >= 1
-            log_string += f'{idx}: '
-            frame = self.dataset.count
-        else:
-            frame = getattr(self.dataset, 'frame', 0)
-
-        self.data_path = p
-        # save_path = str(self.save_dir / p.name)  # im.jpg
-        self.txt_path = str(self.save_dir / 'labels' / p.stem) + ('' if self.dataset.mode == 'image' else f'_{frame}')
-        log_string += '%gx%g ' % im.shape[2:]  # print string
-
-        result = results[idx]
-        if len(result) == 0:
-            return log_string
-        prob = result.probs
-        # Print results
-        n5 = min(len(self.model.names), 5)
-        top5i = prob.argsort(0, descending=True)[:n5].tolist()  # top 5 indices
-        log_string += f"{', '.join(f'{self.model.names[j]} {prob[j]:.2f}' for j in top5i)}, "
-
-        # write
-        if self.args.save or self.args.show:  # Add bbox to image
-            self.plotted_img = result.plot()
-        if self.args.save_txt:  # Write to file
-            text = '\n'.join(f'{prob[j]:.2f} {self.model.names[j]}' for j in top5i)
-            with open(f'{self.txt_path}.txt', 'a') as f:
-                f.write(text + '\n')
-
-        return log_string
-
 
 def predict(cfg=DEFAULT_CFG, use_python=False):
     model = cfg.model or 'yolov8n-cls.pt'  # or "resnet18"

ultralytics/yolo/v8/detect/predict.py

@@ -5,7 +5,6 @@ import torch
 from ultralytics.yolo.engine.predictor import BasePredictor
 from ultralytics.yolo.engine.results import Results
 from ultralytics.yolo.utils import DEFAULT_CFG, ROOT, ops
-from ultralytics.yolo.utils.plotting import save_one_box
 
 
 class DetectionPredictor(BasePredictor):
@@ -34,48 +33,6 @@ class DetectionPredictor(BasePredictor):
             results.append(Results(orig_img=orig_img, path=img_path, names=self.model.names, boxes=pred))
         return results
 
-    def write_results(self, idx, results, batch):
-        p, im, im0 = batch
-        log_string = ''
-        if len(im.shape) == 3:
-            im = im[None]  # expand for batch dim
-        self.seen += 1
-        imc = im0.copy() if self.args.save_crop else im0
-        if self.source_type.webcam or self.source_type.from_img:  # batch_size >= 1
-            log_string += f'{idx}: '
-            frame = self.dataset.count
-        else:
-            frame = getattr(self.dataset, 'frame', 0)
-        self.data_path = p
-        self.txt_path = str(self.save_dir / 'labels' / p.stem) + ('' if self.dataset.mode == 'image' else f'_{frame}')
-        log_string += '%gx%g ' % im.shape[2:]  # print string
-
-        result = results[idx]  # TODO: make boxes inherit from tensors
-        if len(result) == 0:
-            return f'{log_string}(no detections), '
-        det = result.boxes
-        for c in det.cls.unique():
-            n = (det.cls == c).sum()  # detections per class
-            log_string += f"{n} {self.model.names[int(c)]}{'s' * (n > 1)}, "
-
-        if self.args.save or self.args.show:  # Add bbox to image
-            self.plotted_img = result.plot(line_width=self.args.line_thickness)
-
-        # write
-        for d in reversed(det):
-            c, conf, id = int(d.cls), float(d.conf), None if d.id is None else int(d.id.item())
-            if self.args.save_txt:  # Write to file
-                line = (c, *d.xywhn.view(-1)) + (conf, ) * self.args.save_conf + (() if id is None else (id, ))
-                with open(f'{self.txt_path}.txt', 'a') as f:
-                    f.write(('%g ' * len(line)).rstrip() % line + '\n')
-            if self.args.save_crop:
-                save_one_box(d.xyxy,
-                             imc,
-                             file=self.save_dir / 'crops' / self.model.names[c] / f'{self.data_path.stem}.jpg',
-                             BGR=True)
-
-        return log_string
-
 
 def predict(cfg=DEFAULT_CFG, use_python=False):
     model = cfg.model or 'yolov8n.pt'

ultralytics/yolo/v8/pose/predict.py

@@ -2,7 +2,6 @@
 
 from ultralytics.yolo.engine.results import Results
 from ultralytics.yolo.utils import DEFAULT_CFG, ROOT, ops
-from ultralytics.yolo.utils.plotting import save_one_box
 from ultralytics.yolo.v8.detect.predict import DetectionPredictor
 
 
@@ -34,50 +33,6 @@ class PosePredictor(DetectionPredictor):
                         keypoints=pred_kpts))
         return results
 
-    def write_results(self, idx, results, batch):
-        p, im, im0 = batch
-        log_string = ''
-        if len(im.shape) == 3:
-            im = im[None]  # expand for batch dim
-        self.seen += 1
-        imc = im0.copy() if self.args.save_crop else im0
-        if self.source_type.webcam or self.source_type.from_img:  # batch_size >= 1
-            log_string += f'{idx}: '
-            frame = self.dataset.count
-        else:
-            frame = getattr(self.dataset, 'frame', 0)
-        self.data_path = p
-        self.txt_path = str(self.save_dir / 'labels' / p.stem) + ('' if self.dataset.mode == 'image' else f'_{frame}')
-        log_string += '%gx%g ' % im.shape[2:]  # print string
-
-        result = results[idx]  # TODO: make boxes inherit from tensors
-        if len(result) == 0:
-            return f'{log_string}(no detections), '
-        det = result.boxes
-        for c in det.cls.unique():
-            n = (det.cls == c).sum()  # detections per class
-            log_string += f"{n} {self.model.names[int(c)]}{'s' * (n > 1)}, "
-
-        if self.args.save or self.args.show:  # Add bbox to image
-            self.plotted_img = result.plot(line_width=self.args.line_thickness, boxes=self.args.boxes)
-
-        # write
-        for j, d in enumerate(reversed(det)):
-            c, conf, id = int(d.cls), float(d.conf), None if d.id is None else int(d.id.item())
-            if self.args.save_txt:  # Write to file
-                kpt = (result[j].keypoints[:, :2] / d.orig_shape[[1, 0]]).reshape(-1).tolist()
-                box = d.xywhn.view(-1).tolist()
-                line = (c, *box, *kpt) + (conf, ) * self.args.save_conf + (() if id is None else (id, ))
-                with open(f'{self.txt_path}.txt', 'a') as f:
-                    f.write(('%g ' * len(line)).rstrip() % line + '\n')
-            if self.args.save_crop:
-                save_one_box(d.xyxy,
-                             imc,
-                             file=self.save_dir / 'crops' / self.model.model.names[c] / f'{self.data_path.stem}.jpg',
-                             BGR=True)
-
-        return log_string
-
 
 def predict(cfg=DEFAULT_CFG, use_python=False):
     model = cfg.model or 'yolov8n-pose.pt'

ultralytics/yolo/v8/segment/predict.py

@@ -4,7 +4,6 @@ import torch
 
 from ultralytics.yolo.engine.results import Results
 from ultralytics.yolo.utils import DEFAULT_CFG, ROOT, ops
-from ultralytics.yolo.utils.plotting import save_one_box
 from ultralytics.yolo.v8.detect.predict import DetectionPredictor
 
 
@@ -40,55 +39,6 @@ class SegmentationPredictor(DetectionPredictor):
                 Results(orig_img=orig_img, path=img_path, names=self.model.names, boxes=pred[:, :6], masks=masks))
         return results
 
-    def write_results(self, idx, results, batch):
-        p, im, im0 = batch
-        log_string = ''
-        if len(im.shape) == 3:
-            im = im[None]  # expand for batch dim
-        self.seen += 1
-        imc = im0.copy() if self.args.save_crop else im0
-        if self.source_type.webcam or self.source_type.from_img:  # batch_size >= 1
-            log_string += f'{idx}: '
-            frame = self.dataset.count
-        else:
-            frame = getattr(self.dataset, 'frame', 0)
-
-        self.data_path = p
-        self.txt_path = str(self.save_dir / 'labels' / p.stem) + ('' if self.dataset.mode == 'image' else f'_{frame}')
-        log_string += '%gx%g ' % im.shape[2:]  # print string
-
-        result = results[idx]
-        if len(result) == 0:
-            return f'{log_string}(no detections), '
-        det, mask = result.boxes, result.masks  # getting tensors TODO: mask mask,box inherit for tensor
-
-        # Print results
-        for c in det.cls.unique():
-            n = (det.cls == c).sum()  # detections per class
-            log_string += f"{n} {self.model.names[int(c)]}{'s' * (n > 1)}, "
-
-        # Mask plotting
-        if self.args.save or self.args.show:
-            im_gpu = torch.as_tensor(im0, dtype=torch.float16, device=mask.masks.device).permute(
-                2, 0, 1).flip(0).contiguous() / 255 if self.args.retina_masks else im[idx]
-            self.plotted_img = result.plot(line_width=self.args.line_thickness, im_gpu=im_gpu, boxes=self.args.boxes)
-
-        # Write results
-        for j, d in enumerate(reversed(det)):
-            c, conf, id = int(d.cls), float(d.conf), None if d.id is None else int(d.id.item())
-            if self.args.save_txt:  # Write to file
-                seg = mask.xyn[len(det) - j - 1].copy().reshape(-1)  # reversed mask.xyn, (n,2) to (n*2)
-                line = (c, *seg) + (conf, ) * self.args.save_conf + (() if id is None else (id, ))
-                with open(f'{self.txt_path}.txt', 'a') as f:
-                    f.write(('%g ' * len(line)).rstrip() % line + '\n')
-            if self.args.save_crop:
-                save_one_box(d.xyxy,
-                             imc,
-                             file=self.save_dir / 'crops' / self.model.names[c] / f'{self.data_path.stem}.jpg',
-                             BGR=True)
-
-        return log_string
-
 
 def predict(cfg=DEFAULT_CFG, use_python=False):
     model = cfg.model or 'yolov8n-seg.pt'

{ultralytics-8.0.72.dist-info → ultralytics-8.0.74.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ultralytics
-Version: 8.0.72
+Version: 8.0.74
 Summary: Ultralytics YOLOv8
 Home-page: https://github.com/ultralytics/ultralytics
 Author: Ultralytics

{ultralytics-8.0.72.dist-info → ultralytics-8.0.74.dist-info}/RECORD

@@ -1,4 +1,4 @@
-ultralytics/__init__.py,sha256=PcqJX2EMhtWFjVO2sWzWIdbtc9Ck_5nUNdN5nJ0flOg,295
+ultralytics/__init__.py,sha256=5utx_3w9cyNMSlGo1rLIhEl5HW2OT419-ibnyP6WpaY,295
 ultralytics/assets/bus.jpg,sha256=wCAZxJecGR63Od3ZRERe9Aja1Weayrb9Ug751DS_vGM,137419
 ultralytics/assets/zidane.jpg,sha256=Ftc4aeMmen1O0A3o6GCDO9FlfBslLpTAw0gnetx7bts,50427
 ultralytics/datasets/Argoverse.yaml,sha256=EOZ3aUOi9mgCVot4TMiUBXVuovIdTAZ1FSOy7G224vA,2750
@@ -18,7 +18,7 @@ ultralytics/datasets/coco8.yaml,sha256=YIlvnkEGEk9KOX9PtwQkr3gLvLY4CzSCMBXouP66W
 ultralytics/datasets/xView.yaml,sha256=5JIVo4LOvkm8XJZ8Vxp-2SZMNG-K6ED9jfjG305-_cY,5177
 ultralytics/hub/__init__.py,sha256=GdFJZLpnurnOQS35bXK0T-7Qv_Oc5xFZbsxgNKHap_0,3421
 ultralytics/hub/auth.py,sha256=WYOYQqGziw5dm6qS7qQ29JcwLHZC6JmjJ9Dn4rtZlSQ,5206
-ultralytics/hub/session.py,sha256=L7tMtWBmvEAs8b6t1P1fUDhIER9v1FHki4LP8aBjAkU,8176
+ultralytics/hub/session.py,sha256=O_hVbv9AIAAm0bsfe29eGTmnluHW0N5xoSxiy1HhQmo,8513
 ultralytics/hub/utils.py,sha256=TR9D2HxTfkQoxXAgniD_xSrJjzFuHESGTN-V3mYShRw,9723
 ultralytics/models/v3/yolov3-spp.yaml,sha256=shxxs9C4fVvdpvZR2DK-PRsu-JSjU7ilI339J280IHI,1434
 ultralytics/models/v3/yolov3-tiny.yaml,sha256=yHN4XyVK6QjbWspEvKm1rzZQVDkKBjwLgdUhGB0h0fc,1135
@@ -35,8 +35,8 @@ ultralytics/models/v8/yolov8.yaml,sha256=Mt5Sza9rvG0OVJ0XhuZDZXPrqGW6pTbesXXPR0J
 ultralytics/nn/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ultralytics/nn/autobackend.py,sha256=zR3gnXOiRJHlGJk1lps0EUIVW5pqmSjfQTeUc77A9lI,24083
 ultralytics/nn/autoshape.py,sha256=oB3unCdTIMlvsheWx_q5PQjGh79uPD2m04lvTB5T5Ik,11839
-ultralytics/nn/modules.py,sha256=DbaF1MtcPO0qHAYQZ_Ha_cHImYY3rnnMFeR8IsEmgkY,20043
-ultralytics/nn/tasks.py,sha256=b48fDKZKosQn389Hnvhqo1h7irJvq8dG_EIYOMqdDFg,26611
+ultralytics/nn/modules.py,sha256=LH3fKZuZAMXG4C8_TPhgJu6oxsBEcnKo-FyoiktoxRA,20062
+ultralytics/nn/tasks.py,sha256=nKcUkLlF_5t5IsJuN-E9hqtaDgo26wjEGCwEtXdbw1o,26619
 ultralytics/tracker/__init__.py,sha256=WAkNFWbZYTMiaK63GeBWYJ-HEcKwXOl2pTtmQTfOI5I,201
 ultralytics/tracker/track.py,sha256=ccAHi2XAS0Yu93kYw6Gy6pPNcGt8-Wp38IZpQbxFKI0,2289
 ultralytics/tracker/cfg/botsort.yaml,sha256=YPqsA_iuLjLC6HxjgWREvaEBFnSRIR1F9q5Cfrmm0Aw,889
@@ -53,24 +53,24 @@ ultralytics/yolo/__init__.py,sha256=GZ9sffmGZl0xK2yjXZFyZroVpdcwmWrvGu6QwnEXY4w,
 ultralytics/yolo/cfg/__init__.py,sha256=qsdjHHNpOubbGtpmYetLvhMb3-dDjtr__fqSYYxFxLw,17582
 ultralytics/yolo/cfg/default.yaml,sha256=p67_nGsp-pu55l2YIa2PuxN7zpUsPhcukxb-ruC2vVo,6237
 ultralytics/yolo/data/__init__.py,sha256=ui2V_756otLaIU1E_AtUflt9KsJta4ylLHt8rbKX-Dw,483
-ultralytics/yolo/data/augment.py,sha256=ikrc-13omwlhqP3-UWnJlfU3RGLOyfMqJZJrvWDKvN4,31119
-ultralytics/yolo/data/base.py,sha256=HadwteNfW8XxAg4I80IUYxhcVG3e3JcqHpc_2dp-vWw,8819
+ultralytics/yolo/data/augment.py,sha256=Xzzhv1wimRAQEPqZWBCEQJ1BZ4kG-iJM7yoeHnrlx-E,31285
+ultralytics/yolo/data/base.py,sha256=k81GO8AGabAVdGv7igWCWLUJ3RMjx6mOmR3ix-9y4vU,8919
 ultralytics/yolo/data/build.py,sha256=WcbtM1eJIUNa3Q0Xx1VVXzq0nxGCpI89eL8UNOoTKCE,8614
 ultralytics/yolo/data/dataset.py,sha256=Aa8ppeCYCnAb6tWVwBnjBtKNj_irpexUs3981bYcUw4,12740
 ultralytics/yolo/data/dataset_wrappers.py,sha256=IyP-MGHGO75LjX9mqL-17rLpT-m9_rHUZfop6W2y_mw,1330
-ultralytics/yolo/data/utils.py,sha256=WyXYJNQLh17kZSO_QSUX0ay2nyhr36-URb58J_ZMtXU,22555
+ultralytics/yolo/data/utils.py,sha256=rnav5ux-B4kz1h2DEA3UxCywZOI59t1UGzL6sHB3Yeo,22728
 ultralytics/yolo/data/dataloaders/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ultralytics/yolo/data/dataloaders/stream_loaders.py,sha256=rkRe98hh_jKIv-bt0tDRqjwbt7dVlKyy3es0FFeIdog,15194
 ultralytics/yolo/data/dataloaders/v5augmentations.py,sha256=1cHLok8gmlgNMsGVIvjLJkkmIwfs2MU4ZJ-PwJEiYyE,17226
 ultralytics/yolo/data/dataloaders/v5loader.py,sha256=vThacCrsXCVtfMiUA1scmEpvjN9ASrdU7Lk_33XFbyI,49861
 ultralytics/yolo/engine/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ultralytics/yolo/engine/exporter.py,sha256=4R8lzOT5YZBN3qDD0f1l_OlwqzRO_ZI0h8l8Wo-3sBs,41854
-ultralytics/yolo/engine/model.py,sha256=5nkLx4GvQXq8x51MtO8E6fOjyKAhfMNaceHMU9vxZ80,17896
-ultralytics/yolo/engine/predictor.py,sha256=X9t6SPvvOm5yk0JmRNmCc2KADy0DDlShjuTo-2jt2Dk,13571
-ultralytics/yolo/engine/results.py,sha256=tTjAP8p_nPopOXlS_hfMllLxHGbvgBsRKaIjwqK0avc,14286
-ultralytics/yolo/engine/trainer.py,sha256=8IfDc97QuSlV_rrEljHk8xOYjx0-jvho_RCmJshlzkE,29864
+ultralytics/yolo/engine/model.py,sha256=vOUAvQRpkB_TnbTYtqwC94OgkcST0WOHgVK2gso-L94,18027
+ultralytics/yolo/engine/predictor.py,sha256=hhOAGuwhvATPZP1I95_TJqKbj1kLs9fiijB5dNH7dRQ,14720
+ultralytics/yolo/engine/results.py,sha256=ete8AsOQ2KhrSj9nyua7H3q1wp5wE8CrGpyT_MY6H60,17367
+ultralytics/yolo/engine/trainer.py,sha256=mzqMI82kJrlAc7oRO1hs9s7M3C3P3M0R9olKvu5sdq8,29850
 ultralytics/yolo/engine/validator.py,sha256=yIFxZ7wDNfuOMnnIDYmQA9UMJsJewg12gpn1bfhR724,10358
-ultralytics/yolo/utils/__init__.py,sha256=Fd0acHw17p7_LHTp8FlK9pAtInuP_-ihi4aJru7kMQM,24262
+ultralytics/yolo/utils/__init__.py,sha256=_f9f6-QL5iYYa2KijkoL-J9KQ8CEQV5oDvjLcGoTZyQ,24272
 ultralytics/yolo/utils/autobatch.py,sha256=I09al9DZbo6aNyf-GS1N1WFBOWC4xmHB2Yt81SvDd1o,3823
 ultralytics/yolo/utils/benchmarks.py,sha256=_fo6TzZ8YLEJOQa7rG9hjjg2XIkKd-NnoiYNas0tB3k,5414
 ultralytics/yolo/utils/checks.py,sha256=_ck03UUn3ffjsfLz_Sb7TZnRDIygx8x5rrW5LhKtauM,14089
@@ -80,7 +80,7 @@ ultralytics/yolo/utils/files.py,sha256=c-ig7zwG67MwnKgoj1FEpAe-bq1iC9wYXjMqRlzqz
 ultralytics/yolo/utils/instance.py,sha256=rMxybRdrilqpWAOfUtH-YQx4wHNj2Y2lR48u2G8V4bk,11394
 ultralytics/yolo/utils/loss.py,sha256=JTgCauhA-gHhYn7IaMhR5UAUVHG_wx3S8YR5XhkRs88,2889
 ultralytics/yolo/utils/metrics.py,sha256=4nsmjk1-Gp5LLaoT4jo7zfox23qcF2drbqGzzI5EoR8,35011
-ultralytics/yolo/utils/ops.py,sha256=SAz0Vc80wgSrHA9yDrT4Cgadi-miaYxByXiTtMOsb7Y,28081
+ultralytics/yolo/utils/ops.py,sha256=1J__rmIoMRmhf0oH0fXUyoqwyxIHSTVBoiwN1f8rkqE,28187
 ultralytics/yolo/utils/plotting.py,sha256=ko9rfChgaH6IGX_sEvRiNnmnbsOG0XYdhB0yGkIwyPU,20739
 ultralytics/yolo/utils/tal.py,sha256=mL-olmUSXUj9bAl1s11K21OopomZif3Q-SwdgBJ6Pvs,10233
 ultralytics/yolo/utils/torch_utils.py,sha256=S8CD1-pPX6k0IGTb6Bue9_syYEftQjayJ7EDIT5l73Y,19694
@@ -88,29 +88,29 @@ ultralytics/yolo/utils/callbacks/__init__.py,sha256=9NgdqFgjXrL5RN8GpyQtu6EQcDff
 ultralytics/yolo/utils/callbacks/base.py,sha256=xa7MrMIcnfalU1iQlny3vWmfeVOA_3JZry-ztvD7Rl4,3654
 ultralytics/yolo/utils/callbacks/clearml.py,sha256=xC54laB6EUUKeWgooA3oJLaRb-FwMER7M4JlDdelMwQ,5286
 ultralytics/yolo/utils/callbacks/comet.py,sha256=h-Rfw50BCdgwbgLdGGHPQIQQMcP8vJo1Hr0z6fm5Uj4,11929
-ultralytics/yolo/utils/callbacks/hub.py,sha256=TdGS9uA7zDU6IP_GWns76VUQxcE3dgXfUMj_qQR-Nd8,3154
+ultralytics/yolo/utils/callbacks/hub.py,sha256=_1Z_IQKDb6wTKugP3uGigG1bFuj5EscwtT-dxmiA-K0,3189
 ultralytics/yolo/utils/callbacks/mlflow.py,sha256=VvpMP4hrF5Tjgx_TNL7BCoo9pd6q-WZaIA6oIkYFDqE,2512
 ultralytics/yolo/utils/callbacks/tensorboard.py,sha256=pCDad52mQSuT5aouEU57izkjGvG2OnAzxdk0mSysoOc,1297
 ultralytics/yolo/v8/__init__.py,sha256=aEhezkHyW1R-wRF3HyagSCfMy6xhCZj6m4bbIAi6HT0,157
 ultralytics/yolo/v8/classify/__init__.py,sha256=lm4bNTAfDS2ZTljojApMqAKDmDTVlFpq-EC5jjbS31Q,390
-ultralytics/yolo/v8/classify/predict.py,sha256=FhRCG4_fiJ2OqQbmtWRNMLs3gn2-1XR4yl3tTwpK7_4,3087
+ultralytics/yolo/v8/classify/predict.py,sha256=DPEmo6YIfEzc_ZZYXnLu0-Ut25J-Go_YQ_Y5HKl0Now,1459
 ultralytics/yolo/v8/classify/train.py,sha256=g3lOght7MGaNKUzIKciPthXv3IV1s5HqErhiyLUz2UM,6388
 ultralytics/yolo/v8/classify/val.py,sha256=Zqy3Cf9vNGxtPGIkQSdvTHoOTabhQGCpn7yRR2688Zw,2583
 ultralytics/yolo/v8/detect/__init__.py,sha256=Z3h2Qu_ebeQFOOrl1OQgpyGkzWBrFJ5ygf7uL8X9Mkc,276
-ultralytics/yolo/v8/detect/predict.py,sha256=mQlY6DGqtWwRpQfwUeYhYPCBDYhCkGoH51ERRiYzLLk,3988
+ultralytics/yolo/v8/detect/predict.py,sha256=hVwN9YTPE8E49xYaeHcP8ziKy4NQYstxY4pfohcqDr0,2005
 ultralytics/yolo/v8/detect/train.py,sha256=-wVP4smHE7q7K8AS91BoyHymM2Q5ePOUEPYtw5E5NxU,9886
 ultralytics/yolo/v8/detect/val.py,sha256=ir_sVvB1CmBGV48T5PtnnvR6nWPT_Bm2P3S1XIJ_24U,12516
 ultralytics/yolo/v8/pose/__init__.py,sha256=Q94UAg9QgQqi8EATtkllJUBV9weg7Uig8CcO0-ybLw0,246
-ultralytics/yolo/v8/pose/predict.py,sha256=kUzsG-Ky1AACHzCTiljpqvTwUcGmZbJtSSAj9c6PDqc,4238
+ultralytics/yolo/v8/pose/predict.py,sha256=U0T3AIB8m8-U9eLT-mLlV90rSS4ka9nAYCMgWx2_OoI,2076
 ultralytics/yolo/v8/pose/train.py,sha256=MqpezV18uN-g8SzNqzfFr9SJvF_54XfDo29cclVDAGY,7145
 ultralytics/yolo/v8/pose/val.py,sha256=EHRU228qeeNY0XHWUzE4FRS_bCkDoIKo3QDsXPiXHa0,10050
 ultralytics/yolo/v8/segment/__init__.py,sha256=1OK2oVv3A-iWPf_V4Jz0uXhBQZyLd7dlI5rAzxu_IvU,294
-ultralytics/yolo/v8/segment/predict.py,sha256=B4pBQBDTABthc2W-wIm46b6otGBbF_GHGZL6-itNRpU,5003
+ultralytics/yolo/v8/segment/predict.py,sha256=6ub-QmT2ndlEMyHb37Imb4mRdpD9fFk-ZeGFHEMPJuU,2613
 ultralytics/yolo/v8/segment/train.py,sha256=-J4FQ6lgFF_cbBUi4rJMhRNWK93SaeNczR4GsMj5uI4,7674
 ultralytics/yolo/v8/segment/val.py,sha256=iKMpDPUGroUG7EtgzFT-DWIiQYMqLMlQ36pe2R58X3E,11922
-ultralytics-8.0.72.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-ultralytics-8.0.72.dist-info/METADATA,sha256=9fsfhD6UUuj8TQPYj5TOpm9zCZ2CQxL62CyTQ0hzRBE,25936
-ultralytics-8.0.72.dist-info/WHEEL,sha256=pkctZYzUS4AYVn6dJ-7367OJZivF2e8RA9b_ZBjif18,92
-ultralytics-8.0.72.dist-info/entry_points.txt,sha256=Ck1F6qKNokeHozQD5pmaFgXHL6dKyC2qCdyXao2e6Yg,103
-ultralytics-8.0.72.dist-info/top_level.txt,sha256=XP49TwiMw4QGsvTLSYiJhz1xF_k7ev5mQ8jJXaXi45Q,12
-ultralytics-8.0.72.dist-info/RECORD,,
+ultralytics-8.0.74.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+ultralytics-8.0.74.dist-info/METADATA,sha256=STsn6pPMlYOqOGDoZtKWBZ5sVipO19UtKoq1JlNHXao,25936
+ultralytics-8.0.74.dist-info/WHEEL,sha256=pkctZYzUS4AYVn6dJ-7367OJZivF2e8RA9b_ZBjif18,92
+ultralytics-8.0.74.dist-info/entry_points.txt,sha256=Ck1F6qKNokeHozQD5pmaFgXHL6dKyC2qCdyXao2e6Yg,103
+ultralytics-8.0.74.dist-info/top_level.txt,sha256=XP49TwiMw4QGsvTLSYiJhz1xF_k7ev5mQ8jJXaXi45Q,12
+ultralytics-8.0.74.dist-info/RECORD,,