add resrgan 8x, allow use 1x and up to 8x extra models, move BSRGAN model, add nearest

This commit is contained in:
victorca25
2022-10-30 12:52:50 +01:00
committed by victorca25
parent 17a2076f72
commit c9bb33dd43
4 changed files with 33 additions and 6 deletions

View File

@@ -85,6 +85,9 @@ def cleanup_models():
src_path = os.path.join(root_path, "ESRGAN")
dest_path = os.path.join(models_path, "ESRGAN")
move_files(src_path, dest_path)
src_path = os.path.join(models_path, "BSRGAN")
dest_path = os.path.join(models_path, "ESRGAN")
move_files(src_path, dest_path, ".pth")
src_path = os.path.join(root_path, "gfpgan")
dest_path = os.path.join(models_path, "GFPGAN")
move_files(src_path, dest_path)