Ma/video2x
1
0
Fork 0
mirror of https://github.com/k4yt3x/video2x.git synced 2026-05-14 14:07:35 +08:00
Code Issues Packages Projects Releases Wiki Activity

fixed temp folders bug

Browse Source
This commit is contained in:
k4yt3x
2019-04-18 14:56:30 -04:00
parent 6538abd6e5
commit 837aca371e
2 changed files with 11 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
bin/upscaler.py
View File

@@ -56,20 +56,22 @@ class Upscaler:
self.image_format = 'png' self.image_format = 'png'
self.preserve_frames = False self.preserve_frames = False
# create temporary folder/directories def create_temp_folders(self):
"""create temporary folder/directories
"""
self.extracted_frames = tempfile.mkdtemp(dir=self.video2x_cache_folder) self.extracted_frames = tempfile.mkdtemp(dir=self.video2x_cache_folder)
Avalon.debug_info('Extracted frames are being saved to: {}'.format(self.extracted_frames)) Avalon.debug_info('Extracted frames are being saved to: {}'.format(self.extracted_frames))
self.upscaled_frames = tempfile.mkdtemp(dir=self.video2x_cache_folder) self.upscaled_frames = tempfile.mkdtemp(dir=self.video2x_cache_folder)
Avalon.debug_info('Upscaled frames are being saved to: {}'.format(self.upscaled_frames)) Avalon.debug_info('Upscaled frames are being saved to: {}'.format(self.upscaled_frames))
def cleanup(self): def cleanup(self):
# delete temp directories when done """delete temp directories when done
# avalon framework cannot be used if python is shutting down """
# therefore, plain print is used
if not self.preserve_frames: if not self.preserve_frames:
for directory in [self.extracted_frames, self.upscaled_frames]: for directory in [self.extracted_frames, self.upscaled_frames]:
try: try:
# avalon framework cannot be used if python is shutting down
# therefore, plain print is used
print('Cleaning up cache directory: {}'.format(directory)) print('Cleaning up cache directory: {}'.format(directory))
shutil.rmtree(directory) shutil.rmtree(directory)
except (OSError, FileNotFoundError): except (OSError, FileNotFoundError):

6
bin/video2x.py
View File

@@ -52,7 +52,7 @@ import tempfile
import time import time
import traceback import traceback
VERSION = '2.7.0' VERSION = '2.7.1'
# each thread might take up to 2.5 GB during initialization. # each thread might take up to 2.5 GB during initialization.
# (system memory, not to be confused with GPU memory) # (system memory, not to be confused with GPU memory)
@@ -313,7 +313,8 @@ try:
upscaler.image_format = image_format upscaler.image_format = image_format
upscaler.preserve_frames = preserve_frames upscaler.preserve_frames = preserve_frames
# run upscaler- # run upscaler
upscaler.create_temp_folders()
upscaler.run() upscaler.run()
upscaler.cleanup() upscaler.cleanup()
@@ -337,6 +338,7 @@ try:
upscaler.preserve_frames = preserve_frames upscaler.preserve_frames = preserve_frames
# run upscaler # run upscaler
upscaler.create_temp_folders()
upscaler.run() upscaler.run()
upscaler.cleanup() upscaler.cleanup()
else: else: