def run_demo()

in demo.py [0:0]

• 32 lines of code
• 1 McCabe index (conditional complexity)

def run_demo():

    net_input = get_net_input(get_test_h36m_sample())

    model_dir = download_model('h36m')

    model, _ = init_model_from_dir(model_dir)
    model.eval()

    preds = model(**net_input)

    # input keypoints
    kp_loc = net_input['kp_loc'][0]

    # predicted 3d keypoints in camera coords
    kp_pred_3d = preds['shape_image_coord'][0]

    sticks = STICKS['h36m']

    # viz = get_visdom_connection()
    im_proj = show_projections(
        kp_loc[None].detach().cpu().numpy(),
        visdom_env='demo_h36m',
        visdom_win='input_keypoints',
        title='input_keypoints',
        cmap__='rainbow',
        markersize=40,
        sticks=sticks,
        stickwidth=2,
    )

    im_proj = Image.fromarray(im_proj)
    im_proj_path = os.path.join(model_dir, 'demo_projections.png')
    print('Saving input keypoints to %s' % im_proj_path)
    im_proj.save(im_proj_path)

    video_path = os.path.join(model_dir, 'demo_shape.mp4')
    rotating_3d_video(kp_pred_3d.detach().cpu(),
                      video_path=video_path,
                      sticks=sticks,
                      title='rotating 3d',
                      cmap='rainbow',
                      visdom_env='demo_h36m',
                      visdom_win='3d_shape',
                      get_frames=7, )