OpenSourceJoker/video2x
3
0
Fork 0
mirror of https://github.com/k4yt3x/video2x.git synced 2025-01-01 10:29:09 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fixing progress bar error due to string case mismatch

Browse Source
This commit is contained in:
k4yt3x 2020-02-22 02:31:39 -05:00
parent 78ded08ed5
commit a2150a8dbc
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/upscaler.py
View File

@ -4,7 +4,7 @@
Name: Video2X Upscaler Name: Video2X Upscaler
Author: K4YT3X Author: K4YT3X
Date Created: December 10, 2018 Date Created: December 10, 2018
Last Modified: December 11, 2019 Last Modified: February 22, 2020
Dev: SAT3LL Dev: SAT3LL
@ -116,7 +116,7 @@ class Upscaler:
# get number of extracted frames # get number of extracted frames
self.total_frames = 0 self.total_frames = 0
for directory in extracted_frames_directories: for directory in extracted_frames_directories:
self.total_frames += len([f for f in directory.iterdir() if str(f)[-4:] == f'.{self.image_format}']) self.total_frames += len([f for f in directory.iterdir() if str(f).lower().endswith(self.image_format.lower())])
with tqdm(total=self.total_frames, ascii=True, desc='Upscaling Progress') as progress_bar: with tqdm(total=self.total_frames, ascii=True, desc='Upscaling Progress') as progress_bar:
@ -127,7 +127,7 @@ class Upscaler:
while not self.progress_bar_exit_signal: while not self.progress_bar_exit_signal:
with contextlib.suppress(FileNotFoundError): with contextlib.suppress(FileNotFoundError):
self.total_frames_upscaled = len([f for f in self.upscaled_frames.iterdir() if str(f)[-4:] == f'.{self.image_format}']) self.total_frames_upscaled = len([f for f in self.upscaled_frames.iterdir() if str(f).lower().endswith(self.image_format.lower())])
delta = self.total_frames_upscaled - previous_cycle_frames delta = self.total_frames_upscaled - previous_cycle_frames
previous_cycle_frames = self.total_frames_upscaled previous_cycle_frames = self.total_frames_upscaled