9afdc3cd79162c505526ee638ac64227964005e8,gluoncv/data/transforms/presets/yolo.py,,load_test,#Any#Any#Any#Any#Any#Any#,13

Before Change

    
    if isinstance(filenames, str):
        filenames = [filenames]
    tensors = []
    origs = []
    for f in filenames:
        img = mx.image.imread(f)
        img = timage.resize_short_within(img, short, max_size, mult_base=stride)
        orig_img = img.asnumpy().astype("uint8")
        img = mx.nd.image.to_tensor(img)
        img = mx.nd.image.normalize(img, mean=mean, std=std)
        tensors.append(img.expand_dims(0))
        origs.append(orig_img)
    if len(tensors) == 1:
        return tensors[0], origs[0]
    return tensors, origs


class YOLO3DefaultTrainTransform(object):
    Default YOLO training transform which includes tons of image augmentations.

After Change

    
    if isinstance(filenames, str):
        filenames = [filenames]
    imgs = [mx.image.imread(f) for f in filenames]
    return transform_test(imgs, short, max_size, stride, mean, std)


class YOLO3DefaultTrainTransform(object):
    Default YOLO training transform which includes tons of image augmentations.

Instances