Spaces:
Running
on
A10G
Running
on
A10G
File size: 3,725 Bytes
63074f5 714bf26 687b293 714bf26 687b293 714bf26 687b293 714bf26 687b293 714bf26 687b293 714bf26 687b293 714bf26 687b293 63074f5 714bf26 63074f5 714bf26 75453c0 714bf26 75453c0 714bf26 75453c0 714bf26 75453c0 714bf26 75453c0 714bf26 63074f5 714bf26 63074f5 714bf26 9adc565 714bf26 9adc565 714bf26 9adc565 714bf26 9adc565 714bf26 63074f5 714bf26 75453c0 529a0a6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 |
import os
# App Canny utils
def edge_path_to_video_path(edge_path):
video_path = edge_path
vid_name = edge_path.split("/")[-1]
if vid_name == "butterfly.mp4":
video_path = "__assets__/canny_videos_mp4/butterfly.mp4"
elif vid_name == "deer.mp4":
video_path = "__assets__/canny_videos_mp4/deer.mp4"
elif vid_name == "fox.mp4":
video_path = "__assets__/canny_videos_mp4/fox.mp4"
elif vid_name == "girl_dancing.mp4":
video_path = "__assets__/canny_videos_mp4/girl_dancing.mp4"
elif vid_name == "girl_turning.mp4":
video_path = "__assets__/canny_videos_mp4/girl_turning.mp4"
elif vid_name == "halloween.mp4":
video_path = "__assets__/canny_videos_mp4/halloween.mp4"
elif vid_name == "santa.mp4":
video_path = "__assets__/canny_videos_mp4/santa.mp4"
assert os.path.isfile(video_path)
return video_path
# App Pose utils
def motion_to_video_path(motion):
videos = [
"__assets__/poses_skeleton_gifs/dance1_corr.mp4",
"__assets__/poses_skeleton_gifs/dance2_corr.mp4",
"__assets__/poses_skeleton_gifs/dance3_corr.mp4",
"__assets__/poses_skeleton_gifs/dance4_corr.mp4",
"__assets__/poses_skeleton_gifs/dance5_corr.mp4"
]
if len(motion.split(" ")) > 1 and motion.split(" ")[1].isnumeric():
id = int(motion.split(" ")[1]) - 1
return videos[id]
else:
return motion
# App Canny Dreambooth utils
def get_video_from_canny_selection(canny_selection):
if canny_selection == "woman1":
input_video_path = "__assets__/db_files_2fps/woman1.mp4"
elif canny_selection == "woman2":
input_video_path = "__assets__/db_files_2fps/woman2.mp4"
elif canny_selection == "man1":
input_video_path = "__assets__/db_files_2fps/man1.mp4"
elif canny_selection == "woman3":
input_video_path = "__assets__/db_files_2fps/woman3.mp4"
else:
input_video_path = canny_selection
assert os.path.isfile(input_video_path)
return input_video_path
def get_model_from_db_selection(db_selection):
if db_selection == "Anime DB":
input_video_path = 'PAIR/text2video-zero-controlnet-canny-anime'
elif db_selection == "Avatar DB":
input_video_path = 'PAIR/text2video-zero-controlnet-canny-avatar'
elif db_selection == "GTA-5 DB":
input_video_path = 'PAIR/text2video-zero-controlnet-canny-gta5'
elif db_selection == "Arcane DB":
input_video_path = 'PAIR/text2video-zero-controlnet-canny-arcane'
else:
input_video_path = db_selection
return input_video_path
def get_db_name_from_id(id):
db_names = ["Anime DB", "Arcane DB", "GTA-5 DB", "Avatar DB"]
return db_names[id]
def get_canny_name_from_id(id):
canny_names = ["woman1", "woman2", "man1", "woman3"]
return canny_names[id]
def logo_name_to_path(name):
logo_paths = {
'Picsart AI Research': '__assets__/pair_watermark.png',
'Text2Video-Zero': '__assets__/t2v-z_watermark.png',
'None': None
}
if name in logo_paths:
return logo_paths[name]
return name
# App Depth utils
def depth_path_to_video_path(edge_path):
video_path = edge_path
vid_name = edge_path.split("/")[-1]
if vid_name == "girl_dancing.mp4":
video_path = "__assets__/depth_videos_mp4/girl_dancing.mp4"
elif vid_name == "halloween.mp4":
video_path = "__assets__/depth_videos_mp4/halloween.mp4"
elif vid_name == "man.mp4":
video_path = "__assets__/depth_videos_mp4/man.mp4"
elif vid_name == "woman.mp4":
video_path = "__assets__/depth_videos_mp4/woman.mp4"
assert os.path.isfile(video_path)
return video_path
|