kekezack
fixo push a master en kekezack/XNet
d5359a8c2e feat(Initial commit):
XNet - Polyp Segmentation with Wavelet-FFT Enhanced SwinUNETR
Features:
- Implementation of Wavelet_FFT_SwinUNETR architecture for medical image segmentation
- Adaptive Wavelet Augmented Enhancer (AWAE) module for multi-scale feature enhancement
- FFT-based frequency domain enhancement via FFTRCAB modules
- Combined Dice + CrossEntropy + IoU loss function for robust training
- Comprehensive data augmentation pipeline with geometric and photometric transforms
- Early stopping mechanism with automatic resume from best checkpoint
- SwanLab experiment tracking integration
- Full evaluation pipeline with metrics: mDice, mIoU, mHD, mHD95
Project Structure:
- train.py: Main training script with configurable hyperparameters
- eval.py: Model evaluation and visualization tool
- lib/model/: Core model architecture (SwinUNETR backbone)
- lib/modules/: Enhancement modules (AWAE, FFTRCAB)
- datasets/: Polyp detection dataset loader with MONAI integration
Technical Highlights:
- Multi-level wavelet decomposition with adaptive attention
- Frequency-domain feature enhancement using FFT
hai 1 mes