Перейти к содержимому

Ссылка для ultralytics/utils/patches.py

Примечание

Этот файл доступен по адресу https://github.com/ultralytics/ ultralytics/blob/main/ ultralytics/utils/patches .py. Если ты обнаружил проблему, пожалуйста, помоги исправить ее, создав Pull Request 🛠️. Спасибо 🙏!



ultralytics.utils.patches.imread(filename, flags=cv2.IMREAD_COLOR)

Прочитай изображение из файла.

Параметры:

Имя Тип Описание По умолчанию
filename str

Путь к файлу, который нужно прочитать.

требуется
flags int

Флаг, который может принимать значения cv2.IMREAD_*. По умолчанию это значение равно cv2.IMREAD_COLOR.

IMREAD_COLOR

Возвращается:

Тип Описание
ndarray

Прочитанное изображение.

Исходный код в ultralytics/utils/patches.py
def imread(filename: str, flags: int = cv2.IMREAD_COLOR):
    """
    Read an image from a file.

    Args:
        filename (str): Path to the file to read.
        flags (int, optional): Flag that can take values of cv2.IMREAD_*. Defaults to cv2.IMREAD_COLOR.

    Returns:
        (np.ndarray): The read image.
    """
    return cv2.imdecode(np.fromfile(filename, np.uint8), flags)



ultralytics.utils.patches.imwrite(filename, img, params=None)

Запиши изображение в файл.

Параметры:

Имя Тип Описание По умолчанию
filename str

Путь к файлу, который нужно записать.

требуется
img ndarray

Изображение для написания.

требуется
params list of ints

Дополнительные параметры. Смотри документацию OpenCV.

None

Возвращается:

Тип Описание
bool

True, если файл был записан, False в противном случае.

Исходный код в ultralytics/utils/patches.py
def imwrite(filename: str, img: np.ndarray, params=None):
    """
    Write an image to a file.

    Args:
        filename (str): Path to the file to write.
        img (np.ndarray): Image to write.
        params (list of ints, optional): Additional parameters. See OpenCV documentation.

    Returns:
        (bool): True if the file was written, False otherwise.
    """
    try:
        cv2.imencode(Path(filename).suffix, img, params)[1].tofile(filename)
        return True
    except Exception:
        return False



ultralytics.utils.patches.imshow(winname, mat)

Отображает изображение в указанном окне.

Параметры:

Имя Тип Описание По умолчанию
winname str

Название окна.

требуется
mat ndarray

Изображение, которое будет показано.

требуется
Исходный код в ultralytics/utils/patches.py
def imshow(winname: str, mat: np.ndarray):
    """
    Displays an image in the specified window.

    Args:
        winname (str): Name of the window.
        mat (np.ndarray): Image to be shown.
    """
    _imshow(winname.encode("unicode_escape").decode(), mat)



ultralytics.utils.patches.torch_save(*args, use_dill=True, **kwargs)

Опционально используй dill для сериализации лямбда-функций там, где pickle этого не делает, добавляя надежность с 3 повторными попытками и экспоненциального ожидания в случае сбоя сохранения.

Параметры:

Имя Тип Описание По умолчанию
*args tuple

Позиционные аргументы для передачи в torch.save.

()
use_dill bool

Нужно ли пытаться использовать dill для сериализации, если она доступна. По умолчанию установлено значение True.

True
**kwargs any

Аргументы ключевых слов, которые нужно передать на torch.save.

{}
Исходный код в ultralytics/utils/patches.py
def torch_save(*args, use_dill=True, **kwargs):
    """
    Optionally use dill to serialize lambda functions where pickle does not, adding robustness with 3 retries and
    exponential standoff in case of save failure.

    Args:
        *args (tuple): Positional arguments to pass to torch.save.
        use_dill (bool): Whether to try using dill for serialization if available. Defaults to True.
        **kwargs (any): Keyword arguments to pass to torch.save.
    """
    try:
        assert use_dill
        import dill as pickle
    except (AssertionError, ImportError):
        import pickle

    if "pickle_module" not in kwargs:
        kwargs["pickle_module"] = pickle

    for i in range(4):  # 3 retries
        try:
            return _torch_save(*args, **kwargs)
        except RuntimeError as e:  # unable to save, possibly waiting for device to flush or antivirus scan
            if i == 3:
                raise e
            time.sleep((2**i) / 2)  # exponential standoff: 0.5s, 1.0s, 2.0s





Created 2023-11-12, Updated 2024-06-02
Authors: glenn-jocher (5), Burhan-Q (1), Laughing-q (1)