Update fsrimagevideoupscaler-cli.py

Possible fix for crash on Windows
This commit is contained in:
Janis Hutz
2023-03-28 08:29:37 +02:00
committed by GitHub
parent b3a83fdbad
commit 0f05d650f0

View File

@@ -10,7 +10,7 @@
import argparse import argparse
import bin.handler import bin.handler
import os import os
import time import multiprocessing
ap = argparse.ArgumentParser( description='FSRImageVideoUpscaler - CLI' ) ap = argparse.ArgumentParser( description='FSRImageVideoUpscaler - CLI' )
ap.add_argument( 'inputfile', help='File path for the video / image to be upscaled' ) ap.add_argument( 'inputfile', help='File path for the video / image to be upscaled' )
@@ -23,7 +23,9 @@ handler = bin.handler.Handler()
go = True; go = True;
if ( os.path.exists( args.outputfile ) ): if __name__ == '__main__':
multiprocessing.freeze_support();
if ( os.path.exists( args.outputfile ) ):
if ( input( 'File already exists. Do you want to replace it? (y/n) ' ).lower() == 'y' ): if ( input( 'File already exists. Do you want to replace it? (y/n) ' ).lower() == 'y' ):
go = True go = True
os.remove( args.outputfile ); os.remove( args.outputfile );
@@ -31,7 +33,7 @@ if ( os.path.exists( args.outputfile ) ):
print( '\nRefusing to Upscale video. Please delete the file or specify another filepath!') print( '\nRefusing to Upscale video. Please delete the file or specify another filepath!')
go = False go = False
if ( go ): if ( go ):
if ( args.scalefactor ): if ( args.scalefactor ):
if ( args.scalefactor[ len(args.scalefactor) -1: ] == 'x' ): if ( args.scalefactor[ len(args.scalefactor) -1: ] == 'x' ):
if ( args.threads != None ): if ( args.threads != None ):