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

Revert "added rich console patch for Google Colab"

Browse Source 
This reverts commit db0b87597d.
This commit is contained in:
k4yt3x
2022-09-04 21:56:40 +00:00
parent db0b87597d
commit 84b730497b
1 changed files with 3 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
video2x/video2x.py
View File

@@ -27,7 +27,7 @@ __ __ _ _ ___ __ __
Name: Video2X Name: Video2X
Creator: K4YT3X Creator: K4YT3X
Date Created: February 24, 2018 Date Created: February 24, 2018
Last Modified: September 4, 2022 Last Modified: August 28, 2022
Editor: BrianPetkovsek Editor: BrianPetkovsek
Last Modified: June 17, 2019 Last Modified: June 17, 2019
@@ -41,7 +41,6 @@ Last Modified: March 23, 2020
import ctypes import ctypes
import math import math
import os
import signal import signal
import sys import sys
import time import time
@@ -165,14 +164,8 @@ class Video2X:
original_stdout = sys.stdout original_stdout = sys.stdout
original_stderr = sys.stderr 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 # create console for rich's Live display
# force_jupyter must be True if the script is running in Colab console = Console()
# 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 # redirect STDOUT and STDERR to console
sys.stdout = FileProxy(console, sys.stdout) sys.stdout = FileProxy(console, sys.stdout)
@@ -440,4 +433,4 @@ class Video2X:
ProcessingMode.INTERPOLATE, ProcessingMode.INTERPOLATE,
processes, processes,
(threshold, algorithm), (threshold, algorithm),
) )