nilesrogoff
/
waifu2x-wrapper
1
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Infer use of second argument

This commit is contained in:
Niles Rogoff 2017-02-10 18:38:48 -05:00
parent 0912e09574
commit 6333f47e70
No known key found for this signature in database
GPG Key ID: B78B908F23430F80
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
upscale.py
View File

@ -5,8 +5,12 @@ null = open(os.devnull, "w")
infile = sys.argv[1]
outfile = ".".join(sys.argv[1].split(".")[:-1]) + "-waifu2x.png"
noise_level = False
if len(sys.argv) > 2:
outfile = sys.argv[2]
if len(sys.argv) == 2:
if sys.argv[2] in ["0", "1", "2"]:
print("Assuming you don't want to write to a file named " + sys.argv[2])
noise_level = int(sys.argv[2])
else:
outfile = sys.argv[2]
if len(sys.argv) > 3:
noise_level = int(sys.argv[3])
if not noise_level or noise_level == 0: