Kornia v0.5.4 发布!新增多个数据增广 Container
本次Kornia 的更新中,主要增加对Canny的支持以及新增了多种图像处理容器(例如,视频容器,图像增广容器)。 例如对于数据增广中的各种常见格式(mask, bbox, keypoints),以及图像变换的inverse:
aug_list = AugmentationSequential(
K.ColorJitter(0.1, 0.1, 0.1, 0.1, p=1.0),
K.RandomAffine(360, [0.1, 0.1], [0.7, 1.2], [30., 50.], p=1.0),
K.RandomPerspective(0.5, p=1.0),
data_keys=["input", "bbox", "keypoints", "mask"], # Just to define the future input here.
return_transform=False,
same_on_batch=False,
)
# forward the operation
out_tensors = aug_list(img_tensor, bbox,...