Ссылка для 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
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
ultralytics.utils.patches.imshow(winname, mat)
Отображает изображение в указанном окне.
Параметры:
Имя | Тип | Описание | По умолчанию |
---|---|---|---|
winname |
str
|
Название окна. |
требуется |
mat |
ndarray
|
Изображение, которое будет показано. |
требуется |
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. |
{}
|