Trait SmplModel

pub trait SmplModel<B: Backend>:
    Send
    + Sync
    + 'static
    + Any
    + DynClone {
    // Required methods
    fn smpl_type(&self) -> SmplType;
    fn gender(&self) -> Gender;
    fn forward(
        &self,
        options: &SmplOptions,
        betas: &Betas,
        pose_raw: &Pose,
        expression: Option<&Expression>,
    ) -> SmplOutputDynamic<B>;
    fn create_body_with_uv(
        &self,
        smpl_output: &SmplOutputDynamic<B>,
    ) -> SmplOutputDynamic<B>;
    fn get_face_model(&self) -> &dyn FaceModel<B>;
    fn betas2verts(&self, betas: &Betas) -> Tensor<B, 2, Float>;
    fn verts2joints(
        &self,
        verts_t_pose: Tensor<B, 2, Float>,
    ) -> Tensor<B, 2, Float>;
    fn compute_pose_correctives(&self, pose: &Pose) -> Tensor<B, 2, Float>;
    fn compute_pose_feature(&self, pose: &Pose) -> Array1<f32>;
    fn apply_pose(
        &self,
        verts_t_pose: &Tensor<B, 2, Float>,
        normals: Option<&Tensor<B, 2, Float>>,
        tangents: Option<&Tensor<B, 2, Float>>,
        joints: &Tensor<B, 2, Float>,
        lbs_weights: &Tensor<B, 2, Float>,
        pose: &Pose,
    ) -> (Tensor<B, 2, Float>, Option<Tensor<B, 2, Float>>, Option<Tensor<B, 2, Float>>, Tensor<B, 2, Float>);
    fn faces(&self) -> &Tensor<B, 2, Int>;
    fn faces_uv(&self) -> &Tensor<B, 2, Int>;
    fn uv(&self) -> &Tensor<B, 2, Float>;
    fn lbs_weights(&self) -> Tensor<B, 2, Float>;
    fn lbs_weights_split(&self) -> Tensor<B, 2, Float>;
    fn idx_split_2_merged(&self) -> Tensor<B, 1, Int>;
    fn idx_split_2_merged_vec(&self) -> &Vec<usize>;
    fn set_pose_dirs(&mut self, posedirs: Tensor<B, 2, Float>);
    fn get_pose_dirs(&self) -> Tensor<B, 2, Float>;
    fn get_expression_dirs(&self) -> Option<Tensor<B, 2, Float>>;
    fn clone_dyn(&self) -> Box<dyn SmplModel<B>>;
    fn as_any(&self) -> &dyn Any;
}

Trait for a Smpl based model. Smpl-rs expects all Smpl models to implement this.

Required Methods

fn smpl_type(&self) -> SmplType

fn gender(&self) -> Gender

fn forward( &self, options: &SmplOptions, betas: &Betas, pose_raw: &Pose, expression: Option<&Expression>, ) -> SmplOutputDynamic<B>

fn create_body_with_uv( &self, smpl_output: &SmplOutputDynamic<B>, ) -> SmplOutputDynamic<B>

fn get_face_model(&self) -> &dyn FaceModel<B>

fn betas2verts(&self, betas: &Betas) -> Tensor<B, 2, Float>

fn verts2joints(&self, verts_t_pose: Tensor<B, 2, Float>) -> Tensor<B, 2, Float>

fn compute_pose_correctives(&self, pose: &Pose) -> Tensor<B, 2, Float>

fn compute_pose_feature(&self, pose: &Pose) -> Array1<f32>

fn apply_pose( &self, verts_t_pose: &Tensor<B, 2, Float>, normals: Option<&Tensor<B, 2, Float>>, tangents: Option<&Tensor<B, 2, Float>>, joints: &Tensor<B, 2, Float>, lbs_weights: &Tensor<B, 2, Float>, pose: &Pose, ) -> (Tensor<B, 2, Float>, Option<Tensor<B, 2, Float>>, Option<Tensor<B, 2, Float>>, Tensor<B, 2, Float>)

fn faces(&self) -> &Tensor<B, 2, Int>

fn faces_uv(&self) -> &Tensor<B, 2, Int>

fn uv(&self) -> &Tensor<B, 2, Float>

fn lbs_weights(&self) -> Tensor<B, 2, Float>

fn lbs_weights_split(&self) -> Tensor<B, 2, Float>

fn idx_split_2_merged(&self) -> Tensor<B, 1, Int>

fn idx_split_2_merged_vec(&self) -> &Vec<usize>

fn set_pose_dirs(&mut self, posedirs: Tensor<B, 2, Float>)

fn get_pose_dirs(&self) -> Tensor<B, 2, Float>

fn get_expression_dirs(&self) -> Option<Tensor<B, 2, Float>>

fn clone_dyn(&self) -> Box<dyn SmplModel<B>>

fn as_any(&self) -> &dyn Any

Trait Implementations

impl<B: Backend> Clone for Box<dyn SmplModel<B>>

fn clone(&self) -> Box<dyn SmplModel<B>>

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

Implementors

impl<B: Backend> SmplModel<B> for SmplXGPU<B>

where B::FloatTensorPrimitive<2>: Sync, B::IntTensorPrimitive<2>: Sync, B::IntTensorPrimitive<1>: Sync, B::QuantizedTensorPrimitive<1>: Sync, B::QuantizedTensorPrimitive<2>: Sync,