Index

A

add() (medusa.containers.results.BatchResults method) AlignCropModel (class in medusa.crop.align_crop) alpha_blend() (medusa.render.image.PytorchRenderer method) animate() (medusa.containers.threeD.Base3D method) (medusa.containers.threeD.Flame3D method)

apply_perspective_projection() (in module medusa.transforms) apply_vertex_mask() (in module medusa.geometry) (medusa.containers.fourD.Data4D method) Arcface (class in medusa.recon.flame.mica.encoders)

B

Base3D (class in medusa.containers.threeD) BaseCropModel (class in medusa.crop.base) BaseDetector (class in medusa.detect.base) baseline_normalize() (medusa.epoch.EpochsArray method) BasicBlock (class in medusa.recon.flame.deca.encoders) batch_rigid_transform() (in module medusa.recon.flame.lbs)

batch_rodrigues() (in module medusa.recon.flame.lbs) BatchResults (class in medusa.containers.results) BboxCropModel (class in medusa.crop.bbox_crop) blend_shapes() (in module medusa.recon.flame.lbs) Bottleneck (class in medusa.recon.flame.deca.encoders) bw_filter() (in module medusa.preproc.filter)

C

change_handedness() (in module medusa.recon.mpipe._transforms) clone() (medusa.render.lights.SphericalHarmonicsLights method) close() (medusa.io.VideoLoader method) (medusa.io.VideoWriter method) (medusa.recon.flame.base.FlameReconModel method) (medusa.recon.mpipe.mpipe.Mediapipe method) coef_ (medusa.analysis.LinearRegression attribute) combine_transform_matrix() (in module medusa.recon.mpipe._transforms) compose_matrix() (in module medusa.transforms) compose_mats() (medusa.containers.fourD.Data4D method) compute_optimal_rotation() (in module medusa.recon.mpipe._transforms)

compute_optimal_scale() (in module medusa.recon.mpipe._transforms) compute_tri_normals() (in module medusa.geometry) compute_vertex_normals() (in module medusa.geometry) concat() (medusa.containers.results.BatchResults method) conv1x1() (in module medusa.recon.flame.mica.encoders) conv3x3() (in module medusa.recon.flame.deca.encoders) (in module medusa.recon.flame.mica.encoders) create_ortho_matrix() (in module medusa.transforms) create_viewport_matrix() (in module medusa.transforms) crop() (medusa.io.VideoLoader method) crop_faces_video() (medusa.crop.base.BaseCropModel method) crop_matrix_to_3d() (in module medusa.transforms)

D

Data4D (class in medusa.containers.fourD) DecaReconModel (class in medusa.recon.flame.deca.recon) decompose_matrix() (in module medusa.transforms) decompose_mats() (medusa.containers.fourD.Data4D method) DetailGenerator (class in medusa.recon.flame.deca.decoders)

detect_faces_video() (medusa.detect.base.BaseDetector method) DEVICE (in module medusa.defaults) diffuse() (medusa.render.lights.SphericalHarmonicsLights method) download_file() (in module medusa.io) draw_landmarks() (medusa.render.image.PytorchRenderer method)

E

embed_points_in_mesh() (in module medusa.transforms) EpochsArray (class in medusa.epoch) estimate_alignment() (in module medusa.preproc.align) estimate_scale() (in module medusa.recon.mpipe._transforms) estimate_similarity_transform() (in module medusa.transforms)

euler_matrix() (in module medusa.transforms) expansion (medusa.recon.flame.deca.encoders.BasicBlock attribute) (medusa.recon.flame.deca.encoders.Bottleneck attribute) (medusa.recon.flame.mica.encoders.IBasicBlock attribute) exponential_smoothing() (medusa.preproc.filter.OneEuroFilter static method)

F

FancyTimer (class in medusa.benchmark) fc_scale (medusa.recon.flame.mica.encoders.IResNet attribute) filter_faces() (in module medusa.tracking) (medusa.containers.fourD.Data4D method) (medusa.containers.results.BatchResults method) fit() (medusa.analysis.LinearRegression method) Flame3D (class in medusa.containers.threeD) flame_generator (in module medusa.containers.threeD) FLAME_MODELS (in module medusa.defaults) flame_path (in module medusa.containers.threeD) FlameLandmark (class in medusa.recon.flame.decoders) FlameReconModel (class in medusa.recon.flame.base) FlameShape (class in medusa.recon.flame.decoders) FlameTex (class in medusa.recon.flame.decoders) FONT (in module medusa.defaults) forward() (medusa.crop.align_crop.AlignCropModel method) (medusa.crop.bbox_crop.BboxCropModel method) (medusa.crop.bbox_crop.InsightfaceBboxCropModel method) (medusa.detect.scrfd.SCRFDetector method) (medusa.detect.scrfd.SCRFDPostproc method) (medusa.detect.yunet.YunetDetector method) (medusa.landmark.retinaface.RetinafaceLandmarkModel method) (medusa.recognize.retinaface.RetinafaceGenderAgeModel method) (medusa.recognize.retinaface.RetinafaceRecognitionModel method) (medusa.recon.flame.deca.decoders.DetailGenerator method) (medusa.recon.flame.deca.encoders.BasicBlock method) (medusa.recon.flame.deca.encoders.Bottleneck method) (medusa.recon.flame.deca.encoders.ResNet method) (medusa.recon.flame.deca.encoders.ResnetEncoder method) (medusa.recon.flame.deca.recon.DecaReconModel method) (medusa.recon.flame.decoders.FlameLandmark method) (medusa.recon.flame.decoders.FlameShape method) (medusa.recon.flame.decoders.FlameTex method) (medusa.recon.flame.mica.encoders.Arcface method) (medusa.recon.flame.mica.encoders.IBasicBlock method) (medusa.recon.flame.mica.encoders.IResNet method) (medusa.recon.flame.mica.encoders.MappingNetwork method) (medusa.recon.flame.mica.encoders.MicaEncoder method) (medusa.recon.flame.mica.recon.MicaReconModel method) (medusa.recon.mpipe.mpipe.Mediapipe method) (medusa.render.image.PytorchRenderer method)

freezer() (medusa.recon.flame.mica.encoders.Arcface method) from_4D() (medusa.containers.threeD.Flame3D class method) (medusa.containers.threeD.Mediapipe3D class method) (medusa.epoch.EpochsArray class method) from_video() (medusa.containers.fourD.Data4D static method)

G

get_cam_mat() (medusa.recon.flame.base.FlameReconModel method) (medusa.recon.flame.mica.recon.MicaReconModel method) (medusa.recon.mpipe.mpipe.Mediapipe method) get_example_data4d() (in module medusa.data.example_data) get_example_image() (in module medusa.data.example_data) get_example_video() (in module medusa.data.example_data) get_external_data_config() (in module medusa.data.template_data) get_face() (medusa.containers.fourD.Data4D method) get_full_pose() (medusa.recon.flame.decoders.FlameShape method)

get_logger() (in module medusa.log) get_metadata() (medusa.io.VideoLoader method) get_rigid_vertices() (in module medusa.data.template_data) get_template_flame() (in module medusa.data.template_data) get_template_mediapipe() (in module medusa.data.template_data) get_tris() (in module medusa.data.template_data) (medusa.recon.flame.base.FlameReconModel method) (medusa.recon.mpipe.mpipe.Mediapipe method) get_vertex_template() (in module medusa.data.template_data)

I

IBasicBlock (class in medusa.recon.flame.mica.encoders) image2world() (in module medusa.recon.mpipe._transforms) InsightfaceBboxCropModel (class in medusa.crop.bbox_crop)

internal_solve_weighted_orthogonal_problem() (in module medusa.recon.mpipe._transforms) IResNet (class in medusa.recon.flame.mica.encoders) is_dense() (medusa.recon.flame.base.FlameReconModel method) iter() (medusa.benchmark.FancyTimer static method)

K

kaiming_leaky_init() (in module medusa.recon.flame.mica.encoders)

L

landmark_weights (in module medusa.recon.mpipe._transforms) lbs() (in module medusa.recon.flame.lbs) LinearRegression (class in medusa.analysis)

load() (medusa.containers.fourD.Data4D class method) load_inputs() (in module medusa.io) load_obj() (in module medusa.io) LOGGER (in module medusa.defaults)

M

main() (in module medusa.data.regenerate_example_recon_data) MappingNetwork (class in medusa.recon.flame.mica.encoders) Mediapipe (class in medusa.recon.mpipe.mpipe) Mediapipe3D (class in medusa.containers.threeD) medusa.analysis module medusa.benchmark module medusa.containers module medusa.containers.fourD module medusa.containers.results module medusa.containers.threeD module medusa.crop module medusa.crop.align_crop module medusa.crop.base module medusa.crop.bbox_crop module medusa.data module medusa.data.example_data module medusa.data.regenerate_example_recon_data module medusa.data.template_data module medusa.defaults module medusa.detect module medusa.detect.base module medusa.detect.scrfd module medusa.detect.yunet module medusa.epoch module medusa.geometry module medusa.io module medusa.landmark module medusa.landmark.retinaface module medusa.log module medusa.onnx module medusa.preproc module medusa.preproc.align module medusa.preproc.filter module medusa.recognize module medusa.recognize.retinaface module medusa.recon module medusa.recon.flame module medusa.recon.flame.base module medusa.recon.flame.deca module medusa.recon.flame.deca.decoders module medusa.recon.flame.deca.encoders module medusa.recon.flame.deca.recon module medusa.recon.flame.decoders module medusa.recon.flame.lbs module medusa.recon.flame.mica module

medusa.recon.flame.mica.encoders module medusa.recon.flame.mica.recon module medusa.recon.mpipe module medusa.recon.mpipe._transforms module medusa.recon.mpipe.mpipe module medusa.recon.recon module medusa.render module medusa.render.image module medusa.render.lights module medusa.render.overlay module medusa.render.video module medusa.tracking module medusa.transforms module MicaEncoder (class in medusa.recon.flame.mica.encoders) MicaReconModel (class in medusa.recon.flame.mica.recon) model (medusa.recon.mpipe.mpipe.Mediapipe attribute) module medusa.analysis medusa.benchmark medusa.containers medusa.containers.fourD medusa.containers.results medusa.containers.threeD medusa.crop medusa.crop.align_crop medusa.crop.base medusa.crop.bbox_crop medusa.data medusa.data.example_data medusa.data.regenerate_example_recon_data medusa.data.template_data medusa.defaults medusa.detect medusa.detect.base medusa.detect.scrfd medusa.detect.yunet medusa.epoch medusa.geometry medusa.io medusa.landmark medusa.landmark.retinaface medusa.log medusa.onnx medusa.preproc medusa.preproc.align medusa.preproc.filter medusa.recognize medusa.recognize.retinaface medusa.recon medusa.recon.flame medusa.recon.flame.base medusa.recon.flame.deca medusa.recon.flame.deca.decoders medusa.recon.flame.deca.encoders medusa.recon.flame.deca.recon medusa.recon.flame.decoders medusa.recon.flame.lbs medusa.recon.flame.mica medusa.recon.flame.mica.encoders medusa.recon.flame.mica.recon medusa.recon.mpipe medusa.recon.mpipe._transforms medusa.recon.mpipe.mpipe medusa.recon.recon medusa.render medusa.render.image medusa.render.lights medusa.render.overlay medusa.render.video medusa.tracking medusa.transforms move_and_rescale_z() (in module medusa.recon.mpipe._transforms)

O

OneEuroFilter (class in medusa.preproc.filter)

OnnxModel (class in medusa.onnx) Overlay (class in medusa.render.overlay)

P

PCF (class in medusa.recon.mpipe._transforms) predict() (medusa.analysis.LinearRegression method) predict_from_range() (medusa.analysis.LinearRegression method) procrustes_landmark_basis (in module medusa.recon.mpipe._transforms)

project_points_from_embedding() (in module medusa.transforms) project_to_68_landmarks() (medusa.containers.fourD.Data4D method) project_xy() (in module medusa.recon.mpipe._transforms) PytorchRenderer (class in medusa.render.image)

R

random() (medusa.containers.threeD.Flame3D class method) RECON_MODELS (in module medusa.defaults) render() (medusa.render.video.VideoRenderer method) resize_with_pad() (in module medusa.transforms) ResNet (class in medusa.recon.flame.deca.encoders)

ResnetEncoder (class in medusa.recon.flame.deca.encoders) RetinafaceGenderAgeModel (class in medusa.recognize.retinaface) RetinafaceLandmarkModel (class in medusa.landmark.retinaface) RetinafaceRecognitionModel (class in medusa.recognize.retinaface) rot_mat_to_euler() (in module medusa.recon.flame.lbs) run() (medusa.onnx.OnnxModel method)

S

save() (medusa.containers.fourD.Data4D method) (medusa.containers.threeD.Base3D method) save_image() (medusa.render.image.PytorchRenderer static method) save_obj() (in module medusa.io) SCRFDetector (class in medusa.detect.scrfd) SCRFDPostproc (class in medusa.detect.scrfd)

smoothing_factor() (medusa.preproc.filter.OneEuroFilter static method) solve_weighted_orthogonal_problem() (in module medusa.recon.mpipe._transforms) sort_faces() (in module medusa.tracking) (medusa.containers.results.BatchResults method) specular() (medusa.render.lights.SphericalHarmonicsLights method) SphericalHarmonicsLights (class in medusa.render.lights)

T

TEMPLATE (in module medusa.crop.align_crop) time() (medusa.benchmark.FancyTimer method) to_4D() (medusa.epoch.EpochsArray method) to_df() (medusa.benchmark.FancyTimer method) to_dict() (medusa.containers.results.BatchResults method)

to_local() (medusa.containers.fourD.Data4D method) to_mne() (medusa.epoch.EpochsArray method) to_rgb() (medusa.render.overlay.Overlay method) to_world() (medusa.containers.fourD.Data4D method) tqdm_log() (in module medusa.log) transform_mat() (in module medusa.recon.flame.lbs)

U

unproject_xy() (in module medusa.recon.mpipe._transforms)

V

vector_norm() (in module medusa.transforms) VideoIterableDataset (class in medusa.io) VideoLoader (class in medusa.io) VideoMapDataset (class in medusa.io)

videorecon() (in module medusa.recon.recon) VideoRenderer (class in medusa.render.video) VideoWriter (class in medusa.io) visualize() (medusa.containers.results.BatchResults method)

W

write() (medusa.io.VideoWriter method)

Y

YunetDetector (class in medusa.detect.yunet)