A Robust Object Segmentation Network for Under Water Scenes. Accepted by ICASSP 2022. [paper]

The configuration file is config.yaml. You can modify this file to change Hyperparameters.

The code is in the scripts directory.

run python train.py to train the model.

run python test.py to obtain the testing results of the model as well as evaluation metrics.

Our pre-trained model: [Pre-trained Model]

Equation 5 should be

where and represent the amplitude and phase components of the Fourier transform.