├── Lr_earlystop.py ├── Run.py ├── __pycache__ ├── Lr_earlystop.cpython-37.pyc ├── analyze.cpython-37.pyc ├── analyze.cpython-39.pyc ├── config.cpython-37.pyc ├── config.cpython-39.pyc ├── data_preprocess.cpython-37.pyc ├── dataset.cpython-37.pyc ├── dataset.cpython-39.pyc ├── evaluate_rain.cpython-37.pyc ├── evaluate_rain2.cpython-37.pyc ├── ordered_easydict.cpython-37.pyc ├── ordered_easydict.cpython-39.pyc ├── predict.cpython-37.pyc ├── predict.cpython-39.pyc ├── show_preds.cpython-37.pyc ├── train_and_test.cpython-37.pyc ├── utils.cpython-37.pyc └── utils.cpython-39.pyc ├── analyze.py ├── config.py ├── data_preprocess.py ├── dataset.py ├── evaluate_rain.py ├── evaluate_rain2.py ├── evaluation.py ├── maskout.py ├── model ├── UnetModel.py ├── UnetModel_bak.py ├── __pycache__ │ ├── UnetModel.cpython-37.pyc │ ├── UnetModel.cpython-39.pyc │ ├── encoder.cpython-37.pyc │ ├── forecaster.cpython-37.pyc │ ├── loss.cpython-37.pyc │ ├── loss.cpython-39.pyc │ ├── makelayers.cpython-37.pyc │ ├── model.cpython-37.pyc │ ├── net_params.cpython-37.pyc │ ├── network_swinir.cpython-37.pyc │ └── trajGRU.cpython-37.pyc └── loss.py ├── ordered_easydict.py ├── predict.py ├── readme.md ├── show_preds.py ├── train_and_test.py ├── train_and_test3_4.py ├── train_and_test5.py └── utils.py /Lr_earlystop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CAMS-AI/P-UNet_ODiceloss/HEAD/Lr_earlystop.py -------------------------------------------------------------------------------- /Run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CAMS-AI/P-UNet_ODiceloss/HEAD/Run.py -------------------------------------------------------------------------------- /__pycache__/Lr_earlystop.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CAMS-AI/P-UNet_ODiceloss/HEAD/__pycache__/Lr_earlystop.cpython-37.pyc -------------------------------------------------------------------------------- /__pycache__/analyze.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CAMS-AI/P-UNet_ODiceloss/HEAD/__pycache__/analyze.cpython-37.pyc -------------------------------------------------------------------------------- /__pycache__/analyze.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CAMS-AI/P-UNet_ODiceloss/HEAD/__pycache__/analyze.cpython-39.pyc -------------------------------------------------------------------------------- /__pycache__/config.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CAMS-AI/P-UNet_ODiceloss/HEAD/__pycache__/config.cpython-37.pyc -------------------------------------------------------------------------------- /__pycache__/config.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CAMS-AI/P-UNet_ODiceloss/HEAD/__pycache__/config.cpython-39.pyc -------------------------------------------------------------------------------- /__pycache__/data_preprocess.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CAMS-AI/P-UNet_ODiceloss/HEAD/__pycache__/data_preprocess.cpython-37.pyc -------------------------------------------------------------------------------- /__pycache__/dataset.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CAMS-AI/P-UNet_ODiceloss/HEAD/__pycache__/dataset.cpython-37.pyc -------------------------------------------------------------------------------- /__pycache__/dataset.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CAMS-AI/P-UNet_ODiceloss/HEAD/__pycache__/dataset.cpython-39.pyc -------------------------------------------------------------------------------- /__pycache__/evaluate_rain.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CAMS-AI/P-UNet_ODiceloss/HEAD/__pycache__/evaluate_rain.cpython-37.pyc -------------------------------------------------------------------------------- /__pycache__/evaluate_rain2.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CAMS-AI/P-UNet_ODiceloss/HEAD/__pycache__/evaluate_rain2.cpython-37.pyc -------------------------------------------------------------------------------- /__pycache__/ordered_easydict.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CAMS-AI/P-UNet_ODiceloss/HEAD/__pycache__/ordered_easydict.cpython-37.pyc -------------------------------------------------------------------------------- /__pycache__/ordered_easydict.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CAMS-AI/P-UNet_ODiceloss/HEAD/__pycache__/ordered_easydict.cpython-39.pyc -------------------------------------------------------------------------------- /__pycache__/predict.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CAMS-AI/P-UNet_ODiceloss/HEAD/__pycache__/predict.cpython-37.pyc -------------------------------------------------------------------------------- /__pycache__/predict.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CAMS-AI/P-UNet_ODiceloss/HEAD/__pycache__/predict.cpython-39.pyc -------------------------------------------------------------------------------- /__pycache__/show_preds.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CAMS-AI/P-UNet_ODiceloss/HEAD/__pycache__/show_preds.cpython-37.pyc -------------------------------------------------------------------------------- /__pycache__/train_and_test.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CAMS-AI/P-UNet_ODiceloss/HEAD/__pycache__/train_and_test.cpython-37.pyc -------------------------------------------------------------------------------- /__pycache__/utils.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CAMS-AI/P-UNet_ODiceloss/HEAD/__pycache__/utils.cpython-37.pyc -------------------------------------------------------------------------------- /__pycache__/utils.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CAMS-AI/P-UNet_ODiceloss/HEAD/__pycache__/utils.cpython-39.pyc -------------------------------------------------------------------------------- /analyze.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CAMS-AI/P-UNet_ODiceloss/HEAD/analyze.py -------------------------------------------------------------------------------- /config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CAMS-AI/P-UNet_ODiceloss/HEAD/config.py -------------------------------------------------------------------------------- /data_preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CAMS-AI/P-UNet_ODiceloss/HEAD/data_preprocess.py -------------------------------------------------------------------------------- /dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CAMS-AI/P-UNet_ODiceloss/HEAD/dataset.py -------------------------------------------------------------------------------- /evaluate_rain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CAMS-AI/P-UNet_ODiceloss/HEAD/evaluate_rain.py -------------------------------------------------------------------------------- /evaluate_rain2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CAMS-AI/P-UNet_ODiceloss/HEAD/evaluate_rain2.py -------------------------------------------------------------------------------- /evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CAMS-AI/P-UNet_ODiceloss/HEAD/evaluation.py -------------------------------------------------------------------------------- /maskout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CAMS-AI/P-UNet_ODiceloss/HEAD/maskout.py -------------------------------------------------------------------------------- /model/UnetModel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CAMS-AI/P-UNet_ODiceloss/HEAD/model/UnetModel.py -------------------------------------------------------------------------------- /model/UnetModel_bak.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CAMS-AI/P-UNet_ODiceloss/HEAD/model/UnetModel_bak.py -------------------------------------------------------------------------------- /model/__pycache__/UnetModel.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CAMS-AI/P-UNet_ODiceloss/HEAD/model/__pycache__/UnetModel.cpython-37.pyc -------------------------------------------------------------------------------- /model/__pycache__/UnetModel.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CAMS-AI/P-UNet_ODiceloss/HEAD/model/__pycache__/UnetModel.cpython-39.pyc -------------------------------------------------------------------------------- /model/__pycache__/encoder.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CAMS-AI/P-UNet_ODiceloss/HEAD/model/__pycache__/encoder.cpython-37.pyc -------------------------------------------------------------------------------- /model/__pycache__/forecaster.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CAMS-AI/P-UNet_ODiceloss/HEAD/model/__pycache__/forecaster.cpython-37.pyc -------------------------------------------------------------------------------- /model/__pycache__/loss.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CAMS-AI/P-UNet_ODiceloss/HEAD/model/__pycache__/loss.cpython-37.pyc -------------------------------------------------------------------------------- /model/__pycache__/loss.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CAMS-AI/P-UNet_ODiceloss/HEAD/model/__pycache__/loss.cpython-39.pyc -------------------------------------------------------------------------------- /model/__pycache__/makelayers.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CAMS-AI/P-UNet_ODiceloss/HEAD/model/__pycache__/makelayers.cpython-37.pyc -------------------------------------------------------------------------------- /model/__pycache__/model.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CAMS-AI/P-UNet_ODiceloss/HEAD/model/__pycache__/model.cpython-37.pyc -------------------------------------------------------------------------------- /model/__pycache__/net_params.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CAMS-AI/P-UNet_ODiceloss/HEAD/model/__pycache__/net_params.cpython-37.pyc -------------------------------------------------------------------------------- /model/__pycache__/network_swinir.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CAMS-AI/P-UNet_ODiceloss/HEAD/model/__pycache__/network_swinir.cpython-37.pyc -------------------------------------------------------------------------------- /model/__pycache__/trajGRU.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CAMS-AI/P-UNet_ODiceloss/HEAD/model/__pycache__/trajGRU.cpython-37.pyc -------------------------------------------------------------------------------- /model/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CAMS-AI/P-UNet_ODiceloss/HEAD/model/loss.py -------------------------------------------------------------------------------- /ordered_easydict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CAMS-AI/P-UNet_ODiceloss/HEAD/ordered_easydict.py -------------------------------------------------------------------------------- /predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CAMS-AI/P-UNet_ODiceloss/HEAD/predict.py -------------------------------------------------------------------------------- /readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CAMS-AI/P-UNet_ODiceloss/HEAD/readme.md -------------------------------------------------------------------------------- /show_preds.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CAMS-AI/P-UNet_ODiceloss/HEAD/show_preds.py -------------------------------------------------------------------------------- /train_and_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CAMS-AI/P-UNet_ODiceloss/HEAD/train_and_test.py -------------------------------------------------------------------------------- /train_and_test3_4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CAMS-AI/P-UNet_ODiceloss/HEAD/train_and_test3_4.py -------------------------------------------------------------------------------- /train_and_test5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CAMS-AI/P-UNet_ODiceloss/HEAD/train_and_test5.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CAMS-AI/P-UNet_ODiceloss/HEAD/utils.py --------------------------------------------------------------------------------