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

added OpenCV file open check

Browse Source
This commit is contained in:
k4yt3x 2022-02-02 15:20:03 +00:00
parent 6b83734382
commit d8aed8065b
1 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
video2x/video2x.py
View File

@ -13,7 +13,7 @@ __ __ _ _ ___ __ __
Name: Video2X Name: Video2X
Creator: K4YT3X Creator: K4YT3X
Date Created: Feb 24, 2018 Date Created: Feb 24, 2018
Last Modified: August 17, 2021 Last Modified: February 2, 2022
Editor: BrianPetkovsek Editor: BrianPetkovsek
Last Modified: June 17, 2019 Last Modified: June 17, 2019
@ -126,8 +126,14 @@ class Video2X:
raise RuntimeError("unable to find video stream") raise RuntimeError("unable to find video stream")
# get total number of frames to be processed # get total number of frames to be processed
total_frames = int(cv2.VideoCapture(str(path)).get(cv2.CAP_PROP_FRAME_COUNT)) capture = cv2.VideoCapture(str(path))
frame_rate = cv2.VideoCapture(str(path)).get(cv2.CAP_PROP_FPS)
# check if file is opened successfully
if not capture.isOpened():
raise RuntimeError("OpenCV has failed to open the input file")
total_frames = int(capture.get(cv2.CAP_PROP_FRAME_COUNT))
frame_rate = capture.get(cv2.CAP_PROP_FPS)
return video_info["width"], video_info["height"], total_frames, frame_rate return video_info["width"], video_info["height"], total_frames, frame_rate
@ -195,7 +201,6 @@ class Video2X:
self.processor_processes.append(process) self.processor_processes.append(process)
# create progress bar # create progress bar
try: try:
# wait for jobs in queue to deplete # wait for jobs in queue to deplete
while self.encoder.is_alive() is True: while self.encoder.is_alive() is True: