| license: mit | |
| library_name: pytorch | |
| tags: [image-segmentation, defect-detection, segmentation-models-pytorch, onnx] | |
| # SurfaceGuard: surface defect segmentation | |
| Four defect types on 128x800 grayscale strips. Final option: **U-Net ResNet-34 + FPN EfficientNet-B3**, flip TTA, per-class post-processing | |
| tuned on a held-out validation fold. Validation score 0.7297 (threshold 0.5: 0.7121). | |
| | option | raw_score | score | fg_dice | fp_score | empty_acc | params_M | | |
| |---|---|---|---|---|---|---| | |
| | Scratch U-Net + U-Net ResNet-34 + FPN EfficientNet-B3 | 0.7099 | 0.7306 | 0.6632 | 0.8861 | 0.8911 | 44.7598 | | |
| | U-Net ResNet-34 + FPN EfficientNet-B3 | 0.7121 | 0.7297 | 0.6578 | 0.8966 | 0.8995 | 36.9086 | | |
| | FPN EfficientNet-B3 | 0.6882 | 0.7280 | 0.6561 | 0.8947 | 0.8981 | 12.4760 | | |
| | Scratch U-Net + FPN EfficientNet-B3 | 0.6965 | 0.7271 | 0.6444 | 0.9211 | 0.9173 | 20.3272 | | |
| | Scratch U-Net + U-Net ResNet-34 | 0.6914 | 0.7108 | 0.6119 | 0.9467 | 0.9319 | 32.2838 | | |
| | U-Net ResNet-34 | 0.6859 | 0.7084 | 0.6165 | 0.9261 | 0.9155 | 24.4326 | | |
| | Scratch U-Net | 0.6259 | 0.6759 | 0.5976 | 0.8571 | 0.8620 | 7.8512 | | |
| Files: `weights/*.pt` (state dict + config), `pp_params.json`, `selection.json`. Input: 1x1x128x800 in [0, 1]. | |
| IIT Madras BS Deep Learning & GenAI project (Atul Kumar, 24f2003045). Demo: https://huggingface.co/spaces/Sawarn/surfaceguard-demo | |