123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566 |
- _base_ = '../mask_rcnn/mask-rcnn_r50_fpn_1x_coco.py'
- albu_train_transforms = [
- dict(
- type='ShiftScaleRotate',
- shift_limit=0.0625,
- scale_limit=0.0,
- rotate_limit=0,
- interpolation=1,
- p=0.5),
- dict(
- type='RandomBrightnessContrast',
- brightness_limit=[0.1, 0.3],
- contrast_limit=[0.1, 0.3],
- p=0.2),
- dict(
- type='OneOf',
- transforms=[
- dict(
- type='RGBShift',
- r_shift_limit=10,
- g_shift_limit=10,
- b_shift_limit=10,
- p=1.0),
- dict(
- type='HueSaturationValue',
- hue_shift_limit=20,
- sat_shift_limit=30,
- val_shift_limit=20,
- p=1.0)
- ],
- p=0.1),
- dict(type='JpegCompression', quality_lower=85, quality_upper=95, p=0.2),
- dict(type='ChannelShuffle', p=0.1),
- dict(
- type='OneOf',
- transforms=[
- dict(type='Blur', blur_limit=3, p=1.0),
- dict(type='MedianBlur', blur_limit=3, p=1.0)
- ],
- p=0.1),
- ]
- train_pipeline = [
- dict(type='LoadImageFromFile', backend_args={{_base_.backend_args}}),
- dict(type='LoadAnnotations', with_bbox=True, with_mask=True),
- dict(type='Resize', scale=(1333, 800), keep_ratio=True),
- dict(
- type='Albu',
- transforms=albu_train_transforms,
- bbox_params=dict(
- type='BboxParams',
- format='pascal_voc',
- label_fields=['gt_bboxes_labels', 'gt_ignore_flags'],
- min_visibility=0.0,
- filter_lost_elements=True),
- keymap={
- 'img': 'image',
- 'gt_masks': 'masks',
- 'gt_bboxes': 'bboxes'
- },
- skip_img_without_anno=True),
- dict(type='RandomFlip', prob=0.5),
- dict(type='PackDetInputs')
- ]
- train_dataloader = dict(dataset=dict(pipeline=train_pipeline))
|