快来给你的宠物视频加个表情特效吧

!export CXX=g++!pip install ninja==1.11.1 ray==2.1.0 plotly==4.14.3 torch==1.10.1 torchvision==0.11.2 moviepy==0.2.3.5 lmdb==0.99

import osimport moxing as mox
if not os.path.exists('gangealing/'):    mox.file.copy_parallel('obs://weilin/gangealing/', 'gangealing/')

cd gangealing/gangealing

model = 'cat'     #@param ['celeba', 'dog', 'cat', 'cub']pic = 'ModelArts.png'video_name = 'demo.mp4'

os.environ['RAW_VIDEO_PATH'] = video_name!chmod 777 ./ffmpegos.environ['FFMPEG_BINARY'] = os.path.join(os.getcwd(), 'ffmpeg')

from pathlib import Pathfrom utils.download import download_model, download_videofrom applications.mixed_reality import run_gangealing_on_videofrom applications import load_stnfrom glob import glob

video_resolution = "512"       #@param [128, 256, 512, 1024, 2048, 4096, 8192]pad_mode = 'center'            #@param ["center", "border"]os.environ['FFMPEG_BINARY'] = os.path.join(os.getcwd(), 'ffmpeg')os.environ['VIDEO_SIZE'] = video_size = str(video_resolution)os.environ['PAD'] = pad_modevideo = Path(os.environ['RAW_VIDEO_PATH']).stemos.environ['FRAME_PATH'] = f'data/video_frames/{video}'os.environ['VIDEO_NAME'] = videovideo_path = f'data/{video}'!chmod 777 process_video.sh!./process_video.sh "$RAW_VIDEO_PATH"

!python prepare_data.py --path "$FRAME_PATH" --out "data/$VIDEO_NAME" --pad "$PAD" --size "$VIDEO_SIZE"

fps = 30batch_size = 1use_flipping = Falsememory_efficient_but_slower = False
if 'cutecat' in video_path:    fps = 60
    class MyDict():     def __init__(self): pass

args = MyDict()args.real_size = int(video_size)args.real_data_path = video_pathargs.fps = fpsargs.batch = batch_sizeargs.transform = ['similarity', 'flow']args.flow_size = 128args.stn_channel_multiplier = 0.5args.num_heads = 1args.distributed = False  # Colab only uses 1 GPUargs.clustering = Falseargs.cluster = Noneargs.objects = Trueargs.no_flip_inference = not use_flippingargs.save_frames = memory_efficient_but_slowerargs.overlay_congealed = Falseargs.ckpt = modelargs.override = Falseargs.out = 'visuals'
if pic == 'dense tracking':    args.label_path = f'assets/masks/{model}_mask.png'    # Feel free to change the parameters below:    args.resolution = 128    args.sigma = 1.3    args.opacity = 0.8    args.objects = Falseelse:  # object lense    args.label_path = f'assets/objects/{model}/{pic}'    args.resolution = 4 * int(video_size)    args.sigma = 0.3    args.opacity = 1.0    args.objects = True
stn = load_stn(args)print('Running Spatial Transformer on frames...')run_gangealing_on_video(args, stn, classifier=None)print('Preparing videos to be displayed...')

from IPython.display import HTMLfrom base64 import b64encode

num = len(list(glob(f'{video}_compressed*')))compressed_name = f'{video}_compressed{num}.mp4'congealed_compressed_name = f'{video}_compressed_congealed{num}.mp4'path = f'visuals/video_{video}/propagated.mp4'congealed_path = f'visuals/video_{video}/congealed.mp4'os.system(f"ffmpeg -i {path} -vcodec libx264 {compressed_name}")os.system(f"ffmpeg -i {congealed_path} -vcodec libx264 {congealed_compressed_name}")

mp4 = open(video_name,'rb').read()data_url = "data:video/mp4;base64," + b64encode(mp4).decode()HTML("""<video width=512 autoplay controls loop><source src="%s" type="video/mp4"></video>""" % (data_url))

mp4_1 = open(compressed_name,'rb').read()data_url_1 = "data:video/mp4;base64," + b64encode(mp4_1).decode()HTML("""<video width=512 autoplay controls loop><source src="%s" type="video/mp4"></video>""" % (data_url_1))