irrmnv/seismic-facies-identification

AIcrowd | Community Prize solution for Seismic Facies Identification challenge

Seismic Facies Identification Challenge

Simple solution (0.857 F1-Score / 0.906 Acc public LB) for take part in AIcrowd Seismic Facies Identification Challenge.

Reference

• Flexible library for training NNs in pytorch
• Explained pipeline from similar competition

For inference

Pretrained weights from one of 5 folds

Solution

• UNet (efficientnet-b3 encoder) + scse
• Augs: HorizontalFlip, ShiftScaleRotate
• Loss: 0.75 * Lovasz + 0.25 * CE
• Optimizer: SGD with LR 0.01, momentum 0.9, weight_decay 1e-4
• Crop-predict: ImageSlicer from pytorch-toolbelt