smpl_core::smpl_x::smpl_x_gpu

Struct SmplXGPU

Source 
pub struct SmplXGPU<B: Backend> {
Show 22 fields
    pub device: B::Device,
    pub smpl_type: SmplType,
    pub gender: Gender,
    pub verts_template: Tensor<B, 2, Float>,
    pub faces: Tensor<B, 2, Int>,
    pub faces_uv_mesh: Tensor<B, 2, Int>,
    pub uv: Tensor<B, 2, Float>,
    pub shape_dirs: Tensor<B, 2, Float>,
    pub expression_dirs: Option<Tensor<B, 2, Float>>,
    pub pose_dirs: Option<Tensor<B, 2, Float>>,
    pub joint_regressor: Tensor<B, 2, Float>,
    pub parent_idx_per_joint: Tensor<B, 1, Int>,
    pub lbs_weights: Tensor<B, 2, Float>,
    pub verts_ones: Tensor<B, 2, Float>,
    pub idx_vuv_2_vnouv: Tensor<B, 1, Int>,
    pub faces_na: DMatrix<u32>,
    pub faces_uv_mesh_na: DMatrix<u32>,
    pub uv_na: DMatrix<f32>,
    pub idx_vuv_2_vnouv_vec: Vec<usize>,
    pub lbs_weights_split: Tensor<B, 2>,
    pub lbs_weights_nd: ArcArray2<f32>,
    pub lbs_weights_split_nd: ArcArray2<f32>,

}

Fields§

§device: B::Device§smpl_type: SmplType§gender: Gender§verts_template: Tensor<B, 2, Float>§faces: Tensor<B, 2, Int>§faces_uv_mesh: Tensor<B, 2, Int>§uv: Tensor<B, 2, Float>§shape_dirs: Tensor<B, 2, Float>§expression_dirs: Option<Tensor<B, 2, Float>>§pose_dirs: Option<Tensor<B, 2, Float>>§joint_regressor: Tensor<B, 2, Float>§parent_idx_per_joint: Tensor<B, 1, Int>§lbs_weights: Tensor<B, 2, Float>§verts_ones: Tensor<B, 2, Float>§idx_vuv_2_vnouv: Tensor<B, 1, Int>§faces_na: DMatrix<u32>§faces_uv_mesh_na: DMatrix<u32>§uv_na: DMatrix<f32>§idx_vuv_2_vnouv_vec: Vec<usize>§lbs_weights_split: Tensor<B, 2>§lbs_weights_nd: ArcArray2<f32>§lbs_weights_split_nd: ArcArray2<f32>

Implementations§

Source§

impl<B: Backend> SmplXGPU<B>

Source

pub fn new_from_matrices( gender: Gender, verts_template: &Array2<f32>, faces: &Array2<u32>, faces_uv_mesh: &Array2<u32>, uv: &Array2<f32>, shape_dirs: &Array3<f32>, expression_dirs: Option<Array3<f32>>, pose_dirs: Option<Array3<f32>>, joint_regressor: &Array2<f32>, parent_idx_per_joint: &Array1<u32>, lbs_weights: Array2<f32>, max_num_betas: usize, max_num_expression_components: usize, ) -> Self

§Panics

Will panic if the matrices don’t match the expected sizes

Source

pub fn new_from_npz( model_path: &str, gender: Gender, max_num_betas: usize, max_num_expression_components: usize, ) -> Self

§Panics

Will panic if the path cannot be opened

Source

pub async fn new_from_npz_async( model_path: &str, gender: Gender, max_num_betas: usize, max_num_expression_components: usize, ) -> Self

§Panics

Will panic if the path cannot be opened Will panic if the translation and rotation do not cover the same number of timesteps

Source

pub fn new_from_reader<R: Read + Seek>( reader: R, gender: Gender, max_num_betas: usize, max_num_expression_components: usize, ) -> Self

§Panics

Will panic if the path cannot be opened Will panic if the translation and rotation do not cover the same number of timesteps

Source

pub fn read_pose_dirs_from_reader<R: Read + Seek>( reader: R, device: &B::Device, ) -> Tensor<B, 2, Float>

§Panics

Will panic if the path cannot be opened Will panic if the translation and rotation do not cover the same number of timesteps

Trait Implementations§

Source§

impl<B: Clone + Backend> Clone for SmplXGPU<B>
where B::Device: Clone,

Source§

fn clone(&self) -> SmplXGPU<B>

Returns a duplicate of the value. Read more
1.0.0 · Source§

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

Performs copy-assignment from source. Read more
Source§

impl<B: Backend> FaceModel<B> for SmplXGPU<B>
where B::IntTensorPrimitive<1>: Sync, B::IntTensorPrimitive<2>: Sync, B::FloatTensorPrimitive<2>: Sync, B::QuantizedTensorPrimitive<2>: Sync,

Source§

fn expression2offsets(&self, expression: &Expression) -> Tensor<B, 2, Float>

Source§

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,

Source§

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

Source§

fn as_any(&self) -> &dyn Any

Source§

fn smpl_type(&self) -> SmplType

Source§

fn gender(&self) -> Gender

Source§

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

Source§

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

Source§

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

Source§

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

Source§

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

Source§

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

Source§

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

Source§

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>)

Source§

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

Source§

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

Source§

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

Source§

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

Source§

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

Source§

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

Source§

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

Source§

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

Source§

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

Source§

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

Auto Trait Implementations§

Blanket Implementations§

Source§

impl<T> AlignerFor<1> for T

Source§

type Aligner = AlignTo1<T>

The AlignTo* type which aligns Self to ALIGNMENT.
Source§

impl<T> AlignerFor<1024> for T

Source§

type Aligner = AlignTo1024<T>

The AlignTo* type which aligns Self to ALIGNMENT.
Source§

impl<T> AlignerFor<128> for T

Source§

type Aligner = AlignTo128<T>

The AlignTo* type which aligns Self to ALIGNMENT.
Source§

impl<T> AlignerFor<16> for T

Source§

type Aligner = AlignTo16<T>

The AlignTo* type which aligns Self to ALIGNMENT.
Source§

impl<T> AlignerFor<16384> for T

Source§

type Aligner = AlignTo16384<T>

The AlignTo* type which aligns Self to ALIGNMENT.
Source§

impl<T> AlignerFor<2> for T

Source§

type Aligner = AlignTo2<T>

The AlignTo* type which aligns Self to ALIGNMENT.
Source§

impl<T> AlignerFor<2048> for T

Source§

type Aligner = AlignTo2048<T>

The AlignTo* type which aligns Self to ALIGNMENT.
Source§

impl<T> AlignerFor<256> for T

Source§

type Aligner = AlignTo256<T>

The AlignTo* type which aligns Self to ALIGNMENT.
Source§

impl<T> AlignerFor<32> for T

Source§

type Aligner = AlignTo32<T>

The AlignTo* type which aligns Self to ALIGNMENT.
Source§

impl<T> AlignerFor<32768> for T

Source§

type Aligner = AlignTo32768<T>

The AlignTo* type which aligns Self to ALIGNMENT.
Source§

impl<T> AlignerFor<4> for T

Source§

type Aligner = AlignTo4<T>

The AlignTo* type which aligns Self to ALIGNMENT.
Source§

impl<T> AlignerFor<4096> for T

Source§

type Aligner = AlignTo4096<T>

The AlignTo* type which aligns Self to ALIGNMENT.
Source§

impl<T> AlignerFor<512> for T

Source§

type Aligner = AlignTo512<T>

The AlignTo* type which aligns Self to ALIGNMENT.
Source§

impl<T> AlignerFor<64> for T

Source§

type Aligner = AlignTo64<T>

The AlignTo* type which aligns Self to ALIGNMENT.
Source§

impl<T> AlignerFor<8> for T

Source§

type Aligner = AlignTo8<T>

The AlignTo* type which aligns Self to ALIGNMENT.
Source§

impl<T> AlignerFor<8192> for T

Source§

type Aligner = AlignTo8192<T>

The AlignTo* type which aligns Self to ALIGNMENT.
Source§

impl<T> Any for T
where T: 'static + ?Sized,

Source§

fn type_id(&self) -> TypeId

Gets the TypeId of self. Read more
Source§

impl<T> Borrow<T> for T
where T: ?Sized,

Source§

fn borrow(&self) -> &T

Immutably borrows from an owned value. Read more
Source§

impl<T> BorrowMut<T> for T
where T: ?Sized,

Source§

fn borrow_mut(&mut self) -> &mut T

Mutably borrows from an owned value. Read more
Source§

impl<T> CloneToUninit for T
where T: Clone,

Source§

unsafe fn clone_to_uninit(&self, dest: *mut u8)

🔬This is a nightly-only experimental API. (clone_to_uninit)
Performs copy-assignment from self to dest. Read more
Source§

impl<T> Downcast<T> for T

Source§

fn downcast(&self) -> &T

Source§

impl<T> DynClone for T
where T: Clone,

Source§

fn __clone_box(&self, _: Private) -> *mut ()

Source§

impl<T> From<T> for T

Source§

fn from(t: T) -> T

Returns the argument unchanged.

Source§

impl<T, U> Into<U> for T
where U: From<T>,

Source§

fn into(self) -> U

Calls U::from(self).

That is, this conversion is whatever the implementation of From<T> for U chooses to do.

Source§

impl<T> IntoEither for T

Source§

fn into_either(self, into_left: bool) -> Either<Self, Self>

Converts self into a Left variant of Either<Self, Self> if into_left is true. Converts self into a Right variant of Either<Self, Self> otherwise. Read more
Source§

fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
where F: FnOnce(&Self) -> bool,

Converts self into a Left variant of Either<Self, Self> if into_left(&self) returns true. Converts self into a Right variant of Either<Self, Self> otherwise. Read more
Source§

impl<T> Pointable for T

Source§

const ALIGN: usize

The alignment of pointer.
Source§

type Init = T

The type for initializers.
Source§

unsafe fn init(init: <T as Pointable>::Init) -> usize

Initializes a with the given initializer. Read more
Source§

unsafe fn deref<'a>(ptr: usize) -> &'a T

Dereferences the given pointer. Read more
Source§

unsafe fn deref_mut<'a>(ptr: usize) -> &'a mut T

Mutably dereferences the given pointer. Read more
Source§

unsafe fn drop(ptr: usize)

Drops the object pointed to by the given pointer. Read more
Source§

impl<'a, T> RCowCompatibleRef<'a> for T
where T: Clone + 'a,

Source§

type RefC = &'a T

The (preferably) ffi-safe equivalent of &Self.
Source§

type ROwned = T

The owned version of Self::RefC.
Source§

fn as_c_ref(from: &'a T) -> <T as RCowCompatibleRef<'a>>::RefC

Converts a reference to an FFI-safe type
Source§

fn as_rust_ref(from: <T as RCowCompatibleRef<'a>>::RefC) -> &'a T

Converts an FFI-safe type to a reference
Source§

impl<S> ROExtAcc for S

Source§

fn f_get<F>(&self, offset: FieldOffset<S, F, Aligned>) -> &F

Gets a reference to a field, determined by offset. Read more
Source§

fn f_get_mut<F>(&mut self, offset: FieldOffset<S, F, Aligned>) -> &mut F

Gets a muatble reference to a field, determined by offset. Read more
Source§

fn f_get_ptr<F, A>(&self, offset: FieldOffset<S, F, A>) -> *const F

Gets a const pointer to a field, the field is determined by offset. Read more
Source§

fn f_get_mut_ptr<F, A>(&mut self, offset: FieldOffset<S, F, A>) -> *mut F

Gets a mutable pointer to a field, determined by offset. Read more
Source§

impl<S> ROExtOps<Aligned> for S

Source§

fn f_replace<F>(&mut self, offset: FieldOffset<S, F, Aligned>, value: F) -> F

Replaces a field (determined by offset) with value, returning the previous value of the field. Read more
Source§

fn f_swap<F>(&mut self, offset: FieldOffset<S, F, Aligned>, right: &mut S)

Swaps a field (determined by offset) with the same field in right. Read more
Source§

fn f_get_copy<F>(&self, offset: FieldOffset<S, F, Aligned>) -> F
where F: Copy,

Gets a copy of a field (determined by offset). The field is determined by offset. Read more
Source§

impl<S> ROExtOps<Unaligned> for S

Source§

fn f_replace<F>(&mut self, offset: FieldOffset<S, F, Unaligned>, value: F) -> F

Replaces a field (determined by offset) with value, returning the previous value of the field. Read more
Source§

fn f_swap<F>(&mut self, offset: FieldOffset<S, F, Unaligned>, right: &mut S)

Swaps a field (determined by offset) with the same field in right. Read more
Source§

fn f_get_copy<F>(&self, offset: FieldOffset<S, F, Unaligned>) -> F
where F: Copy,

Gets a copy of a field (determined by offset). The field is determined by offset. Read more
Source§

impl<T> Same for T

Source§

type Output = T

Should always be Self
Source§

impl<T> SelfOps for T
where T: ?Sized,

Source§

fn eq_id(&self, other: &Self) -> bool

Compares the address of self with the address of other. Read more
Source§

fn piped<F, U>(self, f: F) -> U
where F: FnOnce(Self) -> U, Self: Sized,

Emulates the pipeline operator, allowing method syntax in more places. Read more
Source§

fn piped_ref<'a, F, U>(&'a self, f: F) -> U
where F: FnOnce(&'a Self) -> U,

The same as piped except that the function takes &Self Useful for functions that take &Self instead of Self. Read more
Source§

fn piped_mut<'a, F, U>(&'a mut self, f: F) -> U
where F: FnOnce(&'a mut Self) -> U,

The same as piped, except that the function takes &mut Self. Useful for functions that take &mut Self instead of Self.
Source§

fn mutated<F>(self, f: F) -> Self
where F: FnOnce(&mut Self), Self: Sized,

Mutates self using a closure taking self by mutable reference, passing it along the method chain. Read more
Source§

fn observe<F>(self, f: F) -> Self
where F: FnOnce(&Self), Self: Sized,

Observes the value of self, passing it along unmodified. Useful in long method chains. Read more
Source§

fn into_<T>(self) -> T
where Self: Into<T>,

Performs a conversion with Into. using the turbofish .into_::<_>() syntax. Read more
Source§

fn as_ref_<T>(&self) -> &T
where Self: AsRef<T>, T: ?Sized,

Performs a reference to reference conversion with AsRef, using the turbofish .as_ref_::<_>() syntax. Read more
Source§

fn as_mut_<T>(&mut self) -> &mut T
where Self: AsMut<T>, T: ?Sized,

Performs a mutable reference to mutable reference conversion with AsMut, using the turbofish .as_mut_::<_>() syntax. Read more
Source§

fn drop_(self)
where Self: Sized,

Drops self using method notation. Alternative to std::mem::drop. Read more
Source§

impl<SS, SP> SupersetOf<SS> for SP
where SS: SubsetOf<SP>,

Source§

fn to_subset(&self) -> Option<SS>

The inverse inclusion map: attempts to construct self from the equivalent element of its superset. Read more
Source§

fn is_in_subset(&self) -> bool

Checks if self is actually part of its subset T (and can be converted to it).
Source§

fn to_subset_unchecked(&self) -> SS

Use with care! Same as self.to_subset but without any property checks. Always succeeds.
Source§

fn from_subset(element: &SS) -> SP

The inclusion map: converts self to the equivalent element of its superset.
Source§

impl<T> ToOwned for T
where T: Clone,

Source§

type Owned = T

The resulting type after obtaining ownership.
Source§

fn to_owned(&self) -> T

Creates owned data from borrowed data, usually by cloning. Read more
Source§

fn clone_into(&self, target: &mut T)

Uses borrowed data to replace owned data, usually by cloning. Read more
Source§

impl<This> TransmuteElement for This
where This: ?Sized,

Source§

unsafe fn transmute_element<T>(self) -> Self::TransmutedPtr
where Self: CanTransmuteElement<T>,

Transmutes the element type of this pointer.. Read more
Source§

impl<T, U> TryFrom<U> for T
where U: Into<T>,

Source§

type Error = Infallible

The type returned in the event of a conversion error.
Source§

fn try_from(value: U) -> Result<T, <T as TryFrom<U>>::Error>

Performs the conversion.
Source§

impl<T, U> TryInto<U> for T
where U: TryFrom<T>,

Source§

type Error = <U as TryFrom<T>>::Error

The type returned in the event of a conversion error.
Source§

fn try_into(self) -> Result<U, <U as TryFrom<T>>::Error>

Performs the conversion.
Source§

impl<T> TypeIdentity for T
where T: ?Sized,

Source§

type Type = T

This is always Self.
Source§

fn into_type(self) -> Self::Type
where Self: Sized, Self::Type: Sized,

Converts a value back to the original type.
Source§

fn as_type(&self) -> &Self::Type

Converts a reference back to the original type.
Source§

fn as_type_mut(&mut self) -> &mut Self::Type

Converts a mutable reference back to the original type.
Source§

fn into_type_box(self: Box<Self>) -> Box<Self::Type>

Converts a box back to the original type.
Source§

fn into_type_arc(this: Arc<Self>) -> Arc<Self::Type>

Converts an Arc back to the original type. Read more
Source§

fn into_type_rc(this: Rc<Self>) -> Rc<Self::Type>

Converts an Rc back to the original type. Read more
Source§

fn from_type(this: Self::Type) -> Self
where Self: Sized, Self::Type: Sized,

Converts a value back to the original type.
Source§

fn from_type_ref(this: &Self::Type) -> &Self

Converts a reference back to the original type.
Source§

fn from_type_mut(this: &mut Self::Type) -> &mut Self

Converts a mutable reference back to the original type.
Source§

fn from_type_box(this: Box<Self::Type>) -> Box<Self>

Converts a box back to the original type.
Source§

fn from_type_arc(this: Arc<Self::Type>) -> Arc<Self>

Converts an Arc back to the original type.
Source§

fn from_type_rc(this: Rc<Self::Type>) -> Rc<Self>

Converts an Rc back to the original type.
Source§

impl<T> Upcast<T> for T

Source§

fn upcast(&self) -> Option<&T>

Source§

impl<V, T> VZip<V> for T
where V: MultiLane<T>,

Source§

fn vzip(self) -> V

Source§

impl<T> ErasedDestructor for T
where T: 'static,

Source§

impl<T> WasmNotSend for T
where T: Send,

Source§

impl<T> WasmNotSendSync for T
where T: WasmNotSend + WasmNotSync,

Source§

impl<T> WasmNotSync for T
where T: Sync,