vhad/body-pose-animation
1
0
Fork 0
mirror of https://github.com/gosticks/body-pose-animation.git synced 2025-10-16 11:45:42 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

clean up

Browse Source
This commit is contained in:
Chilcone 2021-02-03 15:53:00 +01:00
parent 0ce8774dab
commit 269a565b87
1 changed files with 4 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
example_fit.py
View File

@ -1,15 +1,8 @@
from modules.camera import SimpleCamera from modules.camera import SimpleCamera
from modules.transform import Transform
from modules.pose import BodyPose, train_pose from modules.pose import BodyPose, train_pose
from renderer import Renderer from renderer import Renderer
import torch
import torchgeometry as tgm
from model import * from model import *
# from renderer import *
from dataset import * from dataset import *
from utils.mapping import *
from utils.general import * from utils.general import *
from camera_estimation import TorchCameraEstimate from camera_estimation import TorchCameraEstimate
@ -46,11 +39,11 @@ ascii_logo = """\
""" """
print(ascii_logo) print(ascii_logo)
conf = load_config() conf = load_config()
print("config loaded")
dataset = SMPLyDataset() dataset = SMPLyDataset()
sample_index = 2 sample_index = 2
samples_dir = conf['inputPath']
rename_files(samples_dir + "/")
sample_transforms = [ sample_transforms = [
[ [
@ -87,7 +80,7 @@ l = SMPLyModel(conf['modelPath'])
model = l.create_model() model = l.create_model()
keypoints, conf = dataset[sample_index] keypoints, conf = dataset[sample_index]
img_path = "./samples/" + str(sample_index) + ".png" img_path = "./samples/" + str(sample_index) + ".png"
print(img_path)
# --------------------------------- # ---------------------------------
# Generate model and get joints # Generate model and get joints
# --------------------------------- # ---------------------------------
@ -98,22 +91,17 @@ joints = model_out.joints.detach().cpu().numpy().squeeze()
# Draw in the joints of interest # Draw in the joints of interest
# --------------------------------- # ---------------------------------
est_scale = estimate_scale(joints, keypoints) est_scale = estimate_scale(joints, keypoints)
print("ESTIMATED SCALE:", est_scale)
# apply scaling to keypoints # apply scaling to keypoints
keypoints = keypoints * est_scale keypoints = keypoints * est_scale
r = Renderer()
# integrating Camera Estimation # integrating Camera Estimation
init_joints = get_torso(joints) init_joints = get_torso(joints)
init_keypoints = get_torso(keypoints) init_keypoints = get_torso(keypoints)
print("image path:", img_path)
# setup renderer # setup renderer
r = Renderer()
r.render_model(model, model_out) r.render_model(model, model_out)
# r.render_joints(joints) # r.render_joints(joints)
# r.render_keypoints(keypoints) # r.render_keypoints(keypoints)
@ -150,7 +138,6 @@ pose, transform, cam_trans = camera.estimate_camera_pos()
camera.setup_visualization(render_points, render_keypoints) camera.setup_visualization(render_points, render_keypoints)
# start renderer # start renderer
# r.start() # r.start()