I used a better UNet to train this，I believe that it can be better in the future.