Holliemin9090 / Mammographic-mass-CAD-via-pseudo-color-mammogram-and-Mask-R-CNN

Fully automatic computer-aided mass detection and segmentation via pseudo-color mammograms and Mask R-CNN
http://arxiv.org/abs/1906.12118
Other
24 stars 9 forks source link

Data Augmentation #9

Open latoa opened 3 years ago

latoa commented 3 years ago

Could you please confirm that data augmentation is duplicating the train and validation sets? Each image is only transformed once, right?

Holliemin9090 commented 3 years ago

Hi, maybe duplicating is not the best decription, since augmented image would be transformed and normally not the same as the original image. The 'imgaug' library was used. Please find more information at https://imgaug.readthedocs.io/en/latest/index.html. The image is augmented once. As for how the augmentation was applied, please check out the code in the original Mask_RCNN page. https://github.com/matterport/Mask_RCNN/blob/master/mrcnn/model.py