b5c563a90f07c86b6939aad71e9506020f8b4eb6,imgaug/augmenters/blur.py,,blur_avg_,#Any#Any#,313

Before Change

    if k_height <= 0 or k_width <= 0 or (k_height, k_width) == (1, 1):
        return image

    iadt.gate_dtypes(
        image,
        allowed=["bool",
                 "uint8", "uint16", "int8", "int16",
                 "float16", "float32", "float64"],
        disallowed=["uint32", "uint64", "uint128", "uint256",
                    "int32", "int64", "int128", "int256",
                    "float96", "float128", "float256"],
        augmenter=None)

    input_dtype = image.dtype
    if image.dtype.name in ["bool", "float16"]:
        image = image.astype(np.float32, copy=False)

After Change

    if input_dtype.kind == "b":
        image_aug = image_aug > 0.5
    elif input_dtype in {iadt._INT8_DTYPE, iadt._FLOAT16_DTYPE}:
        image_aug = iadt.restore_dtypes_(image_aug, input_dtype)

    return image_aug

Instances