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, keypoints, mask)
# Inverse the operation
out_tensor_inv = aug_list.inverse(*out_tensor)
此外,我们新增了PatchAugmentSequential来支持更多最新的图像增强方法。
[0.5.4] - 2021-06-11
Added
- Add Canny edge detection (#1020)
- Added Batched forward function (#1058)
- Added denormalize homography function (#1061)
- Added more augmentations containers (#1014)
- Added calibration module and Undistort 2D points function (#1026)
- Added patch augmentation container (#1095)
Fixed
- Remove lena (#1059) :)

