OpenSourceJoker/video2x
3
0
Fork 0
mirror of https://github.com/k4yt3x/video2x.git synced 2024-12-28 23:19:11 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fixed directory cleaning errors, fixed resolution errors

Browse Source
This commit is contained in:
k4yt3x 2019-03-26 16:17:39 -04:00
parent 4bd9541a32
commit c5a7129415
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
bin/upscaler.py
View File

@ -75,7 +75,7 @@ class Upscaler:
for directory in [self.extracted_frames, self.upscaled_frames]: for directory in [self.extracted_frames, self.upscaled_frames]:
try: try:
print('Cleaning up cache directory: {}'.format()) print('Cleaning up cache directory: {}'.format(directory))
shutil.rmtree(directory) shutil.rmtree(directory)
except (OSError, FileNotFoundError): except (OSError, FileNotFoundError):
pass pass
@ -290,8 +290,8 @@ class Upscaler:
if self.scale_ratio: if self.scale_ratio:
coded_width = video_info['streams'][video_stream_index]['coded_width'] coded_width = video_info['streams'][video_stream_index]['coded_width']
coded_height = video_info['streams'][video_stream_index]['coded_height'] coded_height = video_info['streams'][video_stream_index]['coded_height']
self.scale_width = self.scale_ratio * coded_width self.scale_width = int(self.scale_ratio * coded_width)
self.scale_height = self.scale_ratio * coded_height self.scale_height = int(self.scale_ratio * coded_height)
# upscale images one by one using waifu2x # upscale images one by one using waifu2x
Avalon.info('Starting to upscale extracted images') Avalon.info('Starting to upscale extracted images')