MCPcopy Create free account

hub / github.com/MStypulkowski/diffused-heads / functions

Functions26 in github.com/MStypulkowski/diffused-heads

↓ 8 callersMethodexpand
(self, arr, dim=4)
diffusion.py:142
↓ 2 callersMethodset_params
(self)
diffusion.py:98
↓ 1 callersFunctioncreate_windowed_sequence
(seqs, snip_length, cut_dim=0, cutting_stride=None, pad_samples=0)
utils.py:191
↓ 1 callersFunctioncut_n_stack
(seq, snip_length, cut_dim=0, cutting_stride=None, pad_samples=0)
utils.py:175
↓ 1 callersFunctionget_audio_emb
(audio_path, checkpoint, device)
utils.py:39
↓ 1 callersMethodget_beta_schedule
(self, max_beta=0.999)
diffusion.py:89
↓ 1 callersFunctionget_id_frame
(path, random=False, resize=128)
utils.py:50
↓ 1 callersMethodget_p_params
(self, xt, timesteps, nn_out)
diffusion.py:60
↓ 1 callersMethodget_q_params
(self, xt, timesteps, eps_pred=None, x0=None)
diffusion.py:72
↓ 1 callersMethodget_spaced_beta
(self)
diffusion.py:133
↓ 1 callersFunctionget_temp_path
(tmp_dir, mode="", ext="")
utils.py:152
↓ 1 callersFunctionload_image_to_torch
(dir)
utils.py:146
↓ 1 callersFunctionmain
(args)
sample.py:9
↓ 1 callersFunctionpad_both_ends
(tensor, left, right, dim=0)
utils.py:164
↓ 1 callersMethodreset
(self)
utils.py:35
↓ 1 callersMethodsample
(self, x_cond, audio_emb, n_audio_motion_embs=2, n_motion_frames=2, motion_channels=3)
diffusion.py:27
↓ 1 callersMethodsample_loop
(self, xT, x_cond, motion_frames, audio_emb)
diffusion.py:48
↓ 1 callersFunctionsave_audio
(path, audio, audio_rate=16000)
utils.py:75
↓ 1 callersFunctionsave_video
(path, video, fps=25, scale=2, audio=None, audio_rate=16000, overlay_pts=None, ffmpeg_experimental=False)
utils.py:85
↓ 1 callersMethodspace
(self, n_timesteps_new)
diffusion.py:114
↓ 1 callersMethodspace_timesteps
(self, n_timesteps, target_timesteps)
diffusion.py:122
↓ 1 callersFunctionswp_extension
(file, ext)
utils.py:160
Method__init__
(self, path)
utils.py:21
Method__init__
( self, nn_backbone, device, n_timesteps=1000, in_channels=3, image_size=128, out_channels=6, motion_t
diffusion.py:9
Methodforward
(self, audio)
utils.py:26
Functionget_motion_transforms
(args)
utils.py:66