File size: 2,513 Bytes
4409449 |
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 |
import numpy as np
from .materials import body_material
# green
# GT_SMPL = body_material(0.009, 0.214, 0.029)
GT_SMPL = body_material(0.035, 0.415, 0.122)
# blue
# GEN_SMPL = body_material(0.022, 0.129, 0.439)
# Blues => cmap(0.87)
# GEN_SMPL = body_material(0.035, 0.322, 0.615)
# Oranges => cmap(0.87)
GEN_SMPL = body_material(0.658, 0.214, 0.0114)
class Meshes:
def __init__(self, data, *, gt, mode, faces_path, canonicalize, always_on_floor, oldrender=True, is_smplx=False, **kwargs):
data = prepare_meshes(data, canonicalize=canonicalize,
always_on_floor=always_on_floor,
is_smplx=is_smplx)
if isinstance(faces_path, str):
self.faces = np.load(faces_path)
else:
self.faces = faces_path
self.data = data
self.mode = mode
self.oldrender = oldrender
self.N = len(data)
self.trajectory = data[:, :, [0, 1]].mean(1)
if gt:
self.mat = GT_SMPL
else:
self.mat = GEN_SMPL
def get_sequence_mat(self, frac):
import matplotlib
# cmap = matplotlib.cm.get_cmap('Blues')
cmap = matplotlib.cm.get_cmap('Oranges')
# begin = 0.60
# end = 0.90
begin = 0.50
end = 0.90
rgbcolor = cmap(begin + (end-begin)*frac)
mat = body_material(*rgbcolor, oldrender=self.oldrender)
return mat
def get_root(self, index):
return self.data[index].mean(0)
def get_mean_root(self):
return self.data.mean((0, 1))
def load_in_blender(self, index, mat):
vertices = self.data[index]
faces = self.faces
name = f"{str(index).zfill(4)}"
from .tools import load_numpy_vertices_into_blender
load_numpy_vertices_into_blender(vertices, faces, name, mat)
return name
def __len__(self):
return self.N
def prepare_meshes(data, canonicalize=True, always_on_floor=False, is_smplx=False):
if canonicalize:
print("No canonicalization for now")
# fitted mesh do not need fixing axis
# fix axis
if is_smplx:
data[..., 1] = - data[..., 1]
# data[..., 0] = - data[..., 0]
# Swap axis (gravity=Z instead of Y)
data = data[..., [2, 0, 1]]
# Remove the floor
data[..., 2] -= data[..., 2].min()
# Put all the body on the floor
if always_on_floor:
data[..., 2] -= data[..., 2].min(1)[:, None]
return data
|