keras-yolo3项目之二：utils.py文件

最新推荐文章于 2025-08-27 13:27:23 发布

原创

最新推荐文章于 2025-08-27 13:27:23 发布 · 749 阅读

0 ·

CC 4.0 BY-SA版权

本文深入探讨了Keras YOLO3项目中utils.py文件的重要作用，详细解析了在模型训练过程中所使用的各种实用功能。

utils.py是模型训练时其他实用功能。

from functools import reduce

from PIL import Image
import numpy as np
from matplotlib.colors import rgb_to_hsv, hsv_to_rgb

def compose(*funcs):
    """
    Compose arbitrarily many functions, evaluated left to right.
    将多个函数进行组合，从左到右进行计算。
    Reference: https://mathieularose.com/function-composition-in-python/
    """
    # return lambda x: reduce(lambda v, f: f(v), funcs, x)
    if funcs:
        return reduce(lambda f, g: lambda *a, **kw: g(f(*a, **kw)), funcs)
    else:
        raise ValueError('Composition of empty sequence not supported.')

def letterbox_image(image, size):
    '''
    resize image with unchanged aspect ratio using padding
    将图片调整为指定大小
    '''
    iw, ih = image.size
    w, h = size
    scale = min(w/iw, h/ih)
    nw = int(iw*scale)
    nh = int(ih*scale)

    image = image.resize((nw,nh), Image.BICUBIC)
    new_image = Image.new('RGB', size, (128,128,128))
    new_image.paste(image, ((w-nw)//2, (h-nh)//2))
    return new_image

def rand(a=0, b=1):
    """
    将生成的符合[0,1]均匀分布的随机数扩充至[a,b]之间
	"""
    return np.random.rand()*(b-a) + a

def get_random_data(annotation_line, input_shape, random=True, max_boxes=20, jitter=