how to register a new augmentation method and specific in config yml?

[lucasjinreal]

Hi, is there existed GridMask augmentation in d2? How should I register it if there isn't one, and specify it in config yml?

[ppwwyyxx]

The tutorial has explained how to use new augmentation: https://detectron2.readthedocs.io/en/latest/tutorials/data_loading.html

We do not support defining new augmentation in yaml file, though this is support if using python config, e.g.

detectron2/configs/new_baselines/mask_rcnn_R_50_FPN_100ep_LSJ.py

Lines 43 to 50 in bb8b293

	dataloader.train.mapper.augmentations = [
	L(T.ResizeScale)(
	min_scale=0.1, max_scale=2.0, target_height=image_size, target_width=image_size
	),
	L(T.FixedSizeCrop)(crop_size=(image_size, image_size)),
	L(T.RandomFlip)(horizontal=True),
	]

[lucasjinreal]

thanks, I solved this.