├── .idea ├── .gitignore ├── .name ├── UDGNet.iml ├── inspectionProfiles │ ├── Project_Default.xml │ └── profiles_settings.xml ├── misc.xml └── modules.xml ├── README.md ├── Test_Decomposition_angle.py ├── dataset └── test │ ├── angle │ └── data │ │ ├── 008_2.npy │ │ ├── 011_3.npy │ │ ├── 107_11.npy │ │ ├── 114_13.npy │ │ ├── 119_15.npy │ │ ├── 125_16.npy │ │ ├── 134_38.npy │ │ ├── 139_42.npy │ │ ├── 13_0.npy │ │ ├── 141_43.npy │ │ ├── 165_46.npy │ │ ├── 166_47.npy │ │ ├── 17_1.npy │ │ ├── 19_2.npy │ │ ├── 20_3.npy │ │ ├── 24_4.npy │ │ ├── 35_4.npy │ │ ├── 35_6.npy │ │ ├── 37_8.npy │ │ ├── 45_9.npy │ │ ├── 49_10.npy │ │ ├── 52_12.npy │ │ ├── 53_13.npy │ │ ├── 61_6.npy │ │ ├── 69_17.npy │ │ ├── 70_18.npy │ │ ├── 72_19.npy │ │ ├── 73_20.npy │ │ ├── 74_7.npy │ │ ├── 75_21.npy │ │ ├── 78_22.npy │ │ ├── 81_25.npy │ │ ├── 82_26.npy │ │ ├── 83_27.npy │ │ ├── 85_28.npy │ │ ├── 95_31.npy │ │ ├── 96_32.npy │ │ ├── f_pic0211_14_17.npy │ │ ├── f_pic9111_19_19.npy │ │ ├── f_pic9111_51_20.npy │ │ ├── f_pic9111_64_21.npy │ │ ├── image_02_48.npy │ │ ├── image_04_49.npy │ │ ├── image_05_50.npy │ │ ├── image_09_51.npy │ │ ├── image_10_52.npy │ │ ├── image_12_53.npy │ │ ├── image_13_54.npy │ │ ├── image_14_55.npy │ │ ├── image_17_56.npy │ │ ├── image_19_57.npy │ │ ├── image_22_58.npy │ │ ├── image_25_59.npy │ │ ├── image_29_60.npy │ │ ├── image_33_62.npy │ │ ├── image_35_63.npy │ │ ├── image_37_64.npy │ │ ├── image_42_67.npy │ │ ├── image_44_68.npy │ │ └── image_46_69.npy │ └── rain │ └── data │ ├── 008_2.png │ ├── 011_3.png │ ├── 107_11.png │ ├── 114_13.png │ ├── 119_15.png │ ├── 125_16.png │ ├── 134_38.png │ ├── 139_42.png │ ├── 13_0.png │ ├── 141_43.png │ ├── 165_46.png │ ├── 166_47.png │ ├── 17_1.png │ ├── 19_2.png │ ├── 20_3.png │ ├── 24_4.png │ ├── 35_4.png │ ├── 35_6.png │ ├── 37_8.png │ ├── 45_9.png │ ├── 49_10.png │ ├── 52_12.png │ ├── 53_13.png │ ├── 61_6.png │ ├── 69_17.png │ ├── 70_18.png │ ├── 72_19.png │ ├── 73_20.png │ ├── 74_7.png │ ├── 75_21.png │ ├── 78_22.png │ ├── 81_25.png │ ├── 82_26.png │ ├── 83_27.png │ ├── 85_28.png │ ├── 95_31.png │ ├── 96_32.png │ ├── f_pic0211_14_17.png │ ├── f_pic9111_19_19.png │ ├── f_pic9111_51_20.png │ ├── f_pic9111_64_21.png │ ├── image_02_48.png │ ├── image_04_49.png │ ├── image_05_50.png │ ├── image_09_51.png │ ├── image_10_52.png │ ├── image_12_53.png │ ├── image_13_54.png │ ├── image_14_55.png │ ├── image_17_56.png │ ├── image_19_57.png │ ├── image_22_58.png │ ├── image_25_59.png │ ├── image_29_60.png │ ├── image_33_62.png │ ├── image_35_63.png │ ├── image_37_64.png │ ├── image_42_67.png │ ├── image_44_68.png │ └── image_46_69.png ├── enviroment.txt ├── lib ├── AffineGridGen.py ├── STN_model.py ├── __pycache__ │ ├── AffineGridGen.cpython-36.pyc │ ├── AffineGridGen.cpython-37.pyc │ ├── STN_model.cpython-36.pyc │ ├── STN_model.cpython-37.pyc │ ├── dataset.cpython-36.pyc │ ├── dataset.cpython-37.pyc │ ├── loss_function.cpython-36.pyc │ ├── loss_function.cpython-37.pyc │ ├── mean_patch.cpython-36.pyc │ ├── model.cpython-36.pyc │ ├── model.cpython-37.pyc │ ├── network.cpython-36.pyc │ ├── network.cpython-37.pyc │ ├── utils.cpython-36.pyc │ └── utils.cpython-37.pyc ├── angle_label.py ├── dataset.py ├── loss_function.py ├── mean_patch.py ├── model.py ├── network.py ├── transform.py └── utils.py ├── output └── real_model │ ├── angle_model_backup.pth │ └── generator_backup.pth ├── results └── real_result.png └── train_Decomposition_angle.py /.idea/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/.idea/.gitignore -------------------------------------------------------------------------------- /.idea/.name: -------------------------------------------------------------------------------- 1 | README.md -------------------------------------------------------------------------------- /.idea/UDGNet.iml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/.idea/UDGNet.iml -------------------------------------------------------------------------------- /.idea/inspectionProfiles/Project_Default.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/.idea/inspectionProfiles/Project_Default.xml -------------------------------------------------------------------------------- /.idea/inspectionProfiles/profiles_settings.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/.idea/inspectionProfiles/profiles_settings.xml -------------------------------------------------------------------------------- /.idea/misc.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/.idea/misc.xml -------------------------------------------------------------------------------- /.idea/modules.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/.idea/modules.xml -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/README.md -------------------------------------------------------------------------------- /Test_Decomposition_angle.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/Test_Decomposition_angle.py -------------------------------------------------------------------------------- /dataset/test/angle/data/008_2.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/angle/data/008_2.npy -------------------------------------------------------------------------------- /dataset/test/angle/data/011_3.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/angle/data/011_3.npy -------------------------------------------------------------------------------- /dataset/test/angle/data/107_11.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/angle/data/107_11.npy -------------------------------------------------------------------------------- /dataset/test/angle/data/114_13.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/angle/data/114_13.npy -------------------------------------------------------------------------------- /dataset/test/angle/data/119_15.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/angle/data/119_15.npy -------------------------------------------------------------------------------- /dataset/test/angle/data/125_16.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/angle/data/125_16.npy -------------------------------------------------------------------------------- /dataset/test/angle/data/134_38.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/angle/data/134_38.npy -------------------------------------------------------------------------------- /dataset/test/angle/data/139_42.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/angle/data/139_42.npy -------------------------------------------------------------------------------- /dataset/test/angle/data/13_0.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/angle/data/13_0.npy -------------------------------------------------------------------------------- /dataset/test/angle/data/141_43.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/angle/data/141_43.npy -------------------------------------------------------------------------------- /dataset/test/angle/data/165_46.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/angle/data/165_46.npy -------------------------------------------------------------------------------- /dataset/test/angle/data/166_47.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/angle/data/166_47.npy -------------------------------------------------------------------------------- /dataset/test/angle/data/17_1.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/angle/data/17_1.npy -------------------------------------------------------------------------------- /dataset/test/angle/data/19_2.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/angle/data/19_2.npy -------------------------------------------------------------------------------- /dataset/test/angle/data/20_3.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/angle/data/20_3.npy -------------------------------------------------------------------------------- /dataset/test/angle/data/24_4.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/angle/data/24_4.npy -------------------------------------------------------------------------------- /dataset/test/angle/data/35_4.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/angle/data/35_4.npy -------------------------------------------------------------------------------- /dataset/test/angle/data/35_6.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/angle/data/35_6.npy -------------------------------------------------------------------------------- /dataset/test/angle/data/37_8.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/angle/data/37_8.npy -------------------------------------------------------------------------------- /dataset/test/angle/data/45_9.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/angle/data/45_9.npy -------------------------------------------------------------------------------- /dataset/test/angle/data/49_10.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/angle/data/49_10.npy -------------------------------------------------------------------------------- /dataset/test/angle/data/52_12.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/angle/data/52_12.npy -------------------------------------------------------------------------------- /dataset/test/angle/data/53_13.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/angle/data/53_13.npy -------------------------------------------------------------------------------- /dataset/test/angle/data/61_6.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/angle/data/61_6.npy -------------------------------------------------------------------------------- /dataset/test/angle/data/69_17.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/angle/data/69_17.npy -------------------------------------------------------------------------------- /dataset/test/angle/data/70_18.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/angle/data/70_18.npy -------------------------------------------------------------------------------- /dataset/test/angle/data/72_19.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/angle/data/72_19.npy -------------------------------------------------------------------------------- /dataset/test/angle/data/73_20.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/angle/data/73_20.npy -------------------------------------------------------------------------------- /dataset/test/angle/data/74_7.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/angle/data/74_7.npy -------------------------------------------------------------------------------- /dataset/test/angle/data/75_21.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/angle/data/75_21.npy -------------------------------------------------------------------------------- /dataset/test/angle/data/78_22.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/angle/data/78_22.npy -------------------------------------------------------------------------------- /dataset/test/angle/data/81_25.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/angle/data/81_25.npy -------------------------------------------------------------------------------- /dataset/test/angle/data/82_26.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/angle/data/82_26.npy -------------------------------------------------------------------------------- /dataset/test/angle/data/83_27.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/angle/data/83_27.npy -------------------------------------------------------------------------------- /dataset/test/angle/data/85_28.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/angle/data/85_28.npy -------------------------------------------------------------------------------- /dataset/test/angle/data/95_31.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/angle/data/95_31.npy -------------------------------------------------------------------------------- /dataset/test/angle/data/96_32.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/angle/data/96_32.npy -------------------------------------------------------------------------------- /dataset/test/angle/data/f_pic0211_14_17.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/angle/data/f_pic0211_14_17.npy -------------------------------------------------------------------------------- /dataset/test/angle/data/f_pic9111_19_19.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/angle/data/f_pic9111_19_19.npy -------------------------------------------------------------------------------- /dataset/test/angle/data/f_pic9111_51_20.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/angle/data/f_pic9111_51_20.npy -------------------------------------------------------------------------------- /dataset/test/angle/data/f_pic9111_64_21.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/angle/data/f_pic9111_64_21.npy -------------------------------------------------------------------------------- /dataset/test/angle/data/image_02_48.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/angle/data/image_02_48.npy -------------------------------------------------------------------------------- /dataset/test/angle/data/image_04_49.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/angle/data/image_04_49.npy -------------------------------------------------------------------------------- /dataset/test/angle/data/image_05_50.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/angle/data/image_05_50.npy -------------------------------------------------------------------------------- /dataset/test/angle/data/image_09_51.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/angle/data/image_09_51.npy -------------------------------------------------------------------------------- /dataset/test/angle/data/image_10_52.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/angle/data/image_10_52.npy -------------------------------------------------------------------------------- /dataset/test/angle/data/image_12_53.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/angle/data/image_12_53.npy -------------------------------------------------------------------------------- /dataset/test/angle/data/image_13_54.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/angle/data/image_13_54.npy -------------------------------------------------------------------------------- /dataset/test/angle/data/image_14_55.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/angle/data/image_14_55.npy -------------------------------------------------------------------------------- /dataset/test/angle/data/image_17_56.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/angle/data/image_17_56.npy -------------------------------------------------------------------------------- /dataset/test/angle/data/image_19_57.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/angle/data/image_19_57.npy -------------------------------------------------------------------------------- /dataset/test/angle/data/image_22_58.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/angle/data/image_22_58.npy -------------------------------------------------------------------------------- /dataset/test/angle/data/image_25_59.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/angle/data/image_25_59.npy -------------------------------------------------------------------------------- /dataset/test/angle/data/image_29_60.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/angle/data/image_29_60.npy -------------------------------------------------------------------------------- /dataset/test/angle/data/image_33_62.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/angle/data/image_33_62.npy -------------------------------------------------------------------------------- /dataset/test/angle/data/image_35_63.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/angle/data/image_35_63.npy -------------------------------------------------------------------------------- /dataset/test/angle/data/image_37_64.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/angle/data/image_37_64.npy -------------------------------------------------------------------------------- /dataset/test/angle/data/image_42_67.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/angle/data/image_42_67.npy -------------------------------------------------------------------------------- /dataset/test/angle/data/image_44_68.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/angle/data/image_44_68.npy -------------------------------------------------------------------------------- /dataset/test/angle/data/image_46_69.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/angle/data/image_46_69.npy -------------------------------------------------------------------------------- /dataset/test/rain/data/008_2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/rain/data/008_2.png -------------------------------------------------------------------------------- /dataset/test/rain/data/011_3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/rain/data/011_3.png -------------------------------------------------------------------------------- /dataset/test/rain/data/107_11.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/rain/data/107_11.png -------------------------------------------------------------------------------- /dataset/test/rain/data/114_13.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/rain/data/114_13.png -------------------------------------------------------------------------------- /dataset/test/rain/data/119_15.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/rain/data/119_15.png -------------------------------------------------------------------------------- /dataset/test/rain/data/125_16.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/rain/data/125_16.png -------------------------------------------------------------------------------- /dataset/test/rain/data/134_38.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/rain/data/134_38.png -------------------------------------------------------------------------------- /dataset/test/rain/data/139_42.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/rain/data/139_42.png -------------------------------------------------------------------------------- /dataset/test/rain/data/13_0.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/rain/data/13_0.png -------------------------------------------------------------------------------- /dataset/test/rain/data/141_43.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/rain/data/141_43.png -------------------------------------------------------------------------------- /dataset/test/rain/data/165_46.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/rain/data/165_46.png -------------------------------------------------------------------------------- /dataset/test/rain/data/166_47.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/rain/data/166_47.png -------------------------------------------------------------------------------- /dataset/test/rain/data/17_1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/rain/data/17_1.png -------------------------------------------------------------------------------- /dataset/test/rain/data/19_2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/rain/data/19_2.png -------------------------------------------------------------------------------- /dataset/test/rain/data/20_3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/rain/data/20_3.png -------------------------------------------------------------------------------- /dataset/test/rain/data/24_4.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/rain/data/24_4.png -------------------------------------------------------------------------------- /dataset/test/rain/data/35_4.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/rain/data/35_4.png -------------------------------------------------------------------------------- /dataset/test/rain/data/35_6.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/rain/data/35_6.png -------------------------------------------------------------------------------- /dataset/test/rain/data/37_8.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/rain/data/37_8.png -------------------------------------------------------------------------------- /dataset/test/rain/data/45_9.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/rain/data/45_9.png -------------------------------------------------------------------------------- /dataset/test/rain/data/49_10.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/rain/data/49_10.png -------------------------------------------------------------------------------- /dataset/test/rain/data/52_12.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/rain/data/52_12.png -------------------------------------------------------------------------------- /dataset/test/rain/data/53_13.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/rain/data/53_13.png -------------------------------------------------------------------------------- /dataset/test/rain/data/61_6.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/rain/data/61_6.png -------------------------------------------------------------------------------- /dataset/test/rain/data/69_17.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/rain/data/69_17.png -------------------------------------------------------------------------------- /dataset/test/rain/data/70_18.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/rain/data/70_18.png -------------------------------------------------------------------------------- /dataset/test/rain/data/72_19.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/rain/data/72_19.png -------------------------------------------------------------------------------- /dataset/test/rain/data/73_20.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/rain/data/73_20.png -------------------------------------------------------------------------------- /dataset/test/rain/data/74_7.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/rain/data/74_7.png -------------------------------------------------------------------------------- /dataset/test/rain/data/75_21.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/rain/data/75_21.png -------------------------------------------------------------------------------- /dataset/test/rain/data/78_22.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/rain/data/78_22.png -------------------------------------------------------------------------------- /dataset/test/rain/data/81_25.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/rain/data/81_25.png -------------------------------------------------------------------------------- /dataset/test/rain/data/82_26.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/rain/data/82_26.png -------------------------------------------------------------------------------- /dataset/test/rain/data/83_27.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/rain/data/83_27.png -------------------------------------------------------------------------------- /dataset/test/rain/data/85_28.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/rain/data/85_28.png -------------------------------------------------------------------------------- /dataset/test/rain/data/95_31.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/rain/data/95_31.png -------------------------------------------------------------------------------- /dataset/test/rain/data/96_32.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/rain/data/96_32.png -------------------------------------------------------------------------------- /dataset/test/rain/data/f_pic0211_14_17.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/rain/data/f_pic0211_14_17.png -------------------------------------------------------------------------------- /dataset/test/rain/data/f_pic9111_19_19.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/rain/data/f_pic9111_19_19.png -------------------------------------------------------------------------------- /dataset/test/rain/data/f_pic9111_51_20.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/rain/data/f_pic9111_51_20.png -------------------------------------------------------------------------------- /dataset/test/rain/data/f_pic9111_64_21.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/rain/data/f_pic9111_64_21.png -------------------------------------------------------------------------------- /dataset/test/rain/data/image_02_48.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/rain/data/image_02_48.png -------------------------------------------------------------------------------- /dataset/test/rain/data/image_04_49.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/rain/data/image_04_49.png -------------------------------------------------------------------------------- /dataset/test/rain/data/image_05_50.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/rain/data/image_05_50.png -------------------------------------------------------------------------------- /dataset/test/rain/data/image_09_51.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/rain/data/image_09_51.png -------------------------------------------------------------------------------- /dataset/test/rain/data/image_10_52.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/rain/data/image_10_52.png -------------------------------------------------------------------------------- /dataset/test/rain/data/image_12_53.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/rain/data/image_12_53.png -------------------------------------------------------------------------------- /dataset/test/rain/data/image_13_54.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/rain/data/image_13_54.png -------------------------------------------------------------------------------- /dataset/test/rain/data/image_14_55.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/rain/data/image_14_55.png -------------------------------------------------------------------------------- /dataset/test/rain/data/image_17_56.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/rain/data/image_17_56.png -------------------------------------------------------------------------------- /dataset/test/rain/data/image_19_57.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/rain/data/image_19_57.png -------------------------------------------------------------------------------- /dataset/test/rain/data/image_22_58.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/rain/data/image_22_58.png -------------------------------------------------------------------------------- /dataset/test/rain/data/image_25_59.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/rain/data/image_25_59.png -------------------------------------------------------------------------------- /dataset/test/rain/data/image_29_60.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/rain/data/image_29_60.png -------------------------------------------------------------------------------- /dataset/test/rain/data/image_33_62.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/rain/data/image_33_62.png -------------------------------------------------------------------------------- /dataset/test/rain/data/image_35_63.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/rain/data/image_35_63.png -------------------------------------------------------------------------------- /dataset/test/rain/data/image_37_64.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/rain/data/image_37_64.png -------------------------------------------------------------------------------- /dataset/test/rain/data/image_42_67.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/rain/data/image_42_67.png -------------------------------------------------------------------------------- /dataset/test/rain/data/image_44_68.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/rain/data/image_44_68.png -------------------------------------------------------------------------------- /dataset/test/rain/data/image_46_69.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/dataset/test/rain/data/image_46_69.png -------------------------------------------------------------------------------- /enviroment.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/enviroment.txt -------------------------------------------------------------------------------- /lib/AffineGridGen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/lib/AffineGridGen.py -------------------------------------------------------------------------------- /lib/STN_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/lib/STN_model.py -------------------------------------------------------------------------------- /lib/__pycache__/AffineGridGen.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/lib/__pycache__/AffineGridGen.cpython-36.pyc -------------------------------------------------------------------------------- /lib/__pycache__/AffineGridGen.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/lib/__pycache__/AffineGridGen.cpython-37.pyc -------------------------------------------------------------------------------- /lib/__pycache__/STN_model.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/lib/__pycache__/STN_model.cpython-36.pyc -------------------------------------------------------------------------------- /lib/__pycache__/STN_model.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/lib/__pycache__/STN_model.cpython-37.pyc -------------------------------------------------------------------------------- /lib/__pycache__/dataset.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/lib/__pycache__/dataset.cpython-36.pyc -------------------------------------------------------------------------------- /lib/__pycache__/dataset.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/lib/__pycache__/dataset.cpython-37.pyc -------------------------------------------------------------------------------- /lib/__pycache__/loss_function.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/lib/__pycache__/loss_function.cpython-36.pyc -------------------------------------------------------------------------------- /lib/__pycache__/loss_function.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/lib/__pycache__/loss_function.cpython-37.pyc -------------------------------------------------------------------------------- /lib/__pycache__/mean_patch.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/lib/__pycache__/mean_patch.cpython-36.pyc -------------------------------------------------------------------------------- /lib/__pycache__/model.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/lib/__pycache__/model.cpython-36.pyc -------------------------------------------------------------------------------- /lib/__pycache__/model.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/lib/__pycache__/model.cpython-37.pyc -------------------------------------------------------------------------------- /lib/__pycache__/network.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/lib/__pycache__/network.cpython-36.pyc -------------------------------------------------------------------------------- /lib/__pycache__/network.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/lib/__pycache__/network.cpython-37.pyc -------------------------------------------------------------------------------- /lib/__pycache__/utils.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/lib/__pycache__/utils.cpython-36.pyc -------------------------------------------------------------------------------- /lib/__pycache__/utils.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/lib/__pycache__/utils.cpython-37.pyc -------------------------------------------------------------------------------- /lib/angle_label.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/lib/angle_label.py -------------------------------------------------------------------------------- /lib/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/lib/dataset.py -------------------------------------------------------------------------------- /lib/loss_function.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/lib/loss_function.py -------------------------------------------------------------------------------- /lib/mean_patch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/lib/mean_patch.py -------------------------------------------------------------------------------- /lib/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/lib/model.py -------------------------------------------------------------------------------- /lib/network.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/lib/network.py -------------------------------------------------------------------------------- /lib/transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/lib/transform.py -------------------------------------------------------------------------------- /lib/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/lib/utils.py -------------------------------------------------------------------------------- /output/real_model/angle_model_backup.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/output/real_model/angle_model_backup.pth -------------------------------------------------------------------------------- /output/real_model/generator_backup.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/output/real_model/generator_backup.pth -------------------------------------------------------------------------------- /results/real_result.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/results/real_result.png -------------------------------------------------------------------------------- /train_Decomposition_angle.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangfengYu-Hust/UDGNet/HEAD/train_Decomposition_angle.py --------------------------------------------------------------------------------