fix dirpath in esrgan script

huanngzh · huanngzh · commit 085a235424ba · 2024-05-26T23:50:02.000+08:00
diff --git a/scripts/real_esrgan.py b/scripts/real_esrgan.py
@@ -183,7 +183,7 @@ def main():
             "pretrained_weights", "Real-ESRGAN", args.model_name + ".pth"
         )
         if not os.path.isfile(model_path):
-            ROOT_DIR = os.path.dirname(os.path.abspath(__file__))
+            ROOT_DIR = os.getcwd()
             for url in file_url:
                 # model_path will be updated
                 model_path = load_file_from_url(

Original file line number	Diff line number	Diff line change
`@@ -183,7 +183,7 @@ def main():`
`183`	`183`	`"pretrained_weights", "Real-ESRGAN", args.model_name + ".pth"`
`184`	`184`	`)`
`185`	`185`	`if not os.path.isfile(model_path):`
`186`		`- ROOT_DIR = os.path.dirname(os.path.abspath(__file__))`
	`186`	`+ ROOT_DIR = os.getcwd()`
`187`	`187`	`for url in file_url:`
`188`	`188`	`# model_path will be updated`
`189`	`189`	`model_path = load_file_from_url(`