انتقل إلى المحتوى

مرجع ل ultralytics/utils/patches.py

ملاحظه

هذا الملف متاح في https://github.com/ultralytics/ultralytics/ نقطة / الرئيسية /ultralytics/ المرافق / التصحيحات.py. إذا اكتشفت مشكلة ، فيرجى المساعدة في إصلاحها من خلال المساهمة في طلب 🛠️ سحب. شكرا لك 🙏!



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

صحيح إذا كان الملف مكتوبا ، خطأ خلاف ذلك.

شفرة المصدر في ultralytics/utils/patches.py
29 30 31 32 33 34 35 363738394041424344 45
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, **kwargs)

استخدم الشبت (إن وجد) لتسلسل وظائف لامدا حيث لا يقوم المخلل بذلك. يضيف أيضا 3 عمليات إعادة محاولة مع المواجهة الأسية في حالة فشل الحفظ لتحسين متانة المشكلات العابرة.

البارامترات:

اسم نوع وصف افتراضي
*args tuple

الحجج الموضعية لتمريرها torch.أنقذ.

()
**kwargs dict

وسيطات الكلمات الأساسية المراد تمريرها إليها torch.أنقذ.

{}
شفرة المصدر في ultralytics/utils/patches.py
63 64 65 66 67 68 69 70 71 72 73 74 75 76 777879 808182 838485 86
def torch_save(*args, **kwargs):
    """
    Use dill (if exists) to serialize the lambda functions where pickle does not do this. Also adds 3 retries with
    exponential standoff in case of save failure to improve robustness to transient issues.

    Args:
        *args (tuple): Positional arguments to pass to torch.save.
        **kwargs (dict): Keyword arguments to pass to torch.save.
    """
    try:
        import dill as pickle  # noqa
    except ImportError:
        import pickle

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

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





تم إنشاء 2023-11-12, اخر تحديث 2023-11-25
المؤلفون: جلين جوشر (3) ، يضحك س (1)