Dataset 首先读取图像、类别信息、坐标信息等,然后对图像做数据增强和预处理,位置信息等也需要进行相应调整。Compose类可以将不同的数据增强方法进行组合并依次进行数组增强操作
Compose类
class Compose(object):
"""将不同的增强方法组合在一起
参数:
transforms (List[Transform]): 进行组合的数据增强方法存储在列表里面
例子:
>>> transforms.Compose([
>>> transforms.CenterCrop(10),
>>> transforms.ToTensor(),
>>> ])
"""
def __init__(self, transforms):
self.transforms = transforms
def __call__(self, img, boxes=None, labels=None):
for t in self.transforms:
img, boxes, labels = t(img, boxes, labels)
return img, boxes, labels
def append(self, transform):
self.transforms.append(transform)
数据增强方法目录
不同数据增强方法及其实现的目录将会在下面更新(更新ing)。