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

added rich console patch for Google Colab

Browse Source
This commit is contained in:
k4yt3x 2022-09-04 21:41:57 +00:00
parent 102340e2be
commit db0b87597d
1 changed files with 10 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
video2x/video2x.py
View File

@ -27,7 +27,7 @@ __ __ _ _ ___ __ __
Name: Video2X
Creator: K4YT3X
Date Created: February 24, 2018
Last Modified: August 28, 2022
Last Modified: September 4, 2022
Editor: BrianPetkovsek
Last Modified: June 17, 2019
@ -41,6 +41,7 @@ Last Modified: March 23, 2020
import ctypes
import math
import os
import signal
import sys
import time
@ -164,8 +165,14 @@ class Video2X:
original_stdout = sys.stdout
original_stderr = sys.stderr
# detect if the program is running in Google Colab
# COLAB_GPU should be set in Colab runtime environments
in_colab = os.environ.get("COLAB_GPU") is not None
# create console for rich's Live display
console = Console()
# force_jupyter must be True if the script is running in Colab
# other wise the console will not print properly
console = Console(force_jupyter=True if in_colab is True else None)
# redirect STDOUT and STDERR to console
sys.stdout = FileProxy(console, sys.stdout)
@ -433,4 +440,4 @@ class Video2X:
ProcessingMode.INTERPOLATE,
processes,
(threshold, algorithm),
)
)