Open MaxTran96 opened 1 year ago
@MaxTran96 I'm curious as well.. I don't see any info in their readme
Hey friends! You can copy this into a Jupyter notebook, preferably Google Colab! Let me know if this helps.
#@title $\color{salmon} {\large \textsf { 📹 RUN REAL-ESRGAN VIDEO UPSCALE PHASE (from video) --CLI📹}}$
import os
import cv2
import time
print("\033[0;34m")
start_time = time.time()
model_name = 'realesr-general-x4v3' #@param ['realesr-general-x4v3', 'realesr-general-wdn-x4v3', 'RealESRGAN_x4plus','RealESRGAN_x4plus_anime_6B','RealESRGAN_x2plus'] {type:"string"}
lowres_vid = '/content/drive/MyDrive/AI/StableWarpFusion/images_out/scared/video/scared(2)__flow.mp4' #@param{type:'string'}
#@markdown $\color{salmon} {\textsf {input your lowres video path here to be upscaled}}$
highres_vid_dir = '/content/drive/MyDrive/' #@param{type:'string'}
#@markdown $\color{salmon} {\textsf {input your high res video output path here}}$
if not os.path.exists(highres_vid_dir):
os.makedirs(highres_vid_dir)
outscale = "4" #@param ['1', '2', '3', '4', '5', '6', '7', '8', '9', '10', '11', '12', '13']
denoising_strength = 1 #@param{type:'slider', step:0.01, min:0.01, max:1}
target_fps = 30 #@param{type:'number'}
fp32 = 'fp32'
gpu_id = 0 #@param ['0', '1', '2', '3', '4']
use_face_enhance = False #@param {type:"boolean"}
if not use_face_enhance:
%cd '/content/Real-ESRGAN/'
!python3 /content/Real-ESRGAN/inference_realesrgan_video.py \
--model_name $model_name \
--outscale $outscale \
-dn $denoising_strength \
--fp32 \
--input $lowres_vid \
--fps $target_fps \
--output $highres_vid_dir
else:
%cd '/content/Real-ESRGAN/'
!python3 /content/Real-ESRGAN/inference_realesrgan_video.py \
--model_name $model_name \
--outscale $outscale \
--face_enhance \
-dn $denoising_strength \
--fp32 \
--input $lowres_vid \
--fps $target_fps \
--output $highres_vid_dir
%cd '/content/'
print("\033[0;34m...Upscaling Phase Has Completed, You May Proceed...")
elapsed_time = time.time() - start_time
print("\033[0;34mElapsed time:", elapsed_time, "seconds")
Given https://github.com/xinntao/Real-ESRGAN/blob/master/inference_realesrgan_video.py
how do i properly call inference_realesrgan_video() on my input video or the folder that contains all the frames of the video