Note

Before

After

Demo projects

attentive gan derainnet with ONNX Runtime in Python

python demo_attentive-gan-derainnet_onnx.py.py

If you want to change the model, specify it with an argument.

    parser.add_argument(
        "--model",
        type=str,
        default='attentive_gan_derainnet_240x360/model_float32.onnx',
    )