Fix memory leak problem in augment

XPixelGroup / BasicSR

Open Source Image and Video Restoration Toolbox for Super-resolution, Denoise, Deblurring, etc. Currently, it includes EDSR, RCAN, SRResNet, SRGAN, ESRGAN, EDVR, BasicVSR, SwinIR, ECBSR, etc. Also support StyleGAN2, DFDNet.

https://basicsr.readthedocs.io/en/latest/

Apache License 2.0

6.71k stars 1.18k forks source link

Fix memory leak problem in augment #676

Open CastellanLiu opened 5 months ago

CastellanLiu commented 5 months ago

In the original implementation of BasicSR, the augment function is implemented via inplace operations, which may cause memory leak problem.