cvtools.data_augs package¶
Subpackages¶
Submodules¶
Module contents¶
-
class
cvtools.data_augs.
Compose
(transforms)[源代码]¶ 基类:
object
Composes several augmentations together. :param transforms: list of transforms to compose. :type transforms: List[Transform]
Example
>>> augmentations.Compose([ >>> transforms.CenterCrop(10), >>> transforms.ToTensor(), >>> ])
-
class
cvtools.data_augs.
RandomSampleCrop
[源代码]¶ 基类:
object
Crop :param img: the image being input during training :type img: Image :param boxes: the original bounding boxes in pt form :type boxes: Tensor :param labels: the class labels for each bbox :type labels: Tensor :param mode: the min and max jaccard overlaps :type mode: float tuple
返回: - (img, boxes, classes)
- img (Image): the cropped image boxes (Tensor): the adjusted bounding boxes in pt form labels (Tensor): the class labels for each bbox