Skip to content

Commit e30d1b0

Browse files
committed
update tutorial
1 parent 13e17b5 commit e30d1b0

2 files changed

Lines changed: 4 additions & 4 deletions

File tree

examples/app_tutorials/tutorial_human_3dpose_estimation_LCN.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2,15 +2,15 @@
22
# -*- coding: utf-8 -*-
33

44
from tensorlayer.app.human_pose_estimation.common import DataReader, visualize_3D_pose, flip_data
5-
from tensorlayer.app.human_pose_estimation.LCN import CGCNN
5+
from tensorlayer.app import computer_vision
66
import numpy as np
77

88
datareader = DataReader()
99
train_data, test_data = datareader.read_2d(which='scale', mode='gt', read_confidence=False)
1010
train_labels, test_labels = datareader.read_3d(which='scale', mode='gt')
11-
network = CGCNN(pretrained=True)
11+
network = computer_vision.human_pose_estimation('3D-pose')
1212
test_data = flip_data(test_data)
13-
result = network(test_data, is_train=False)
13+
result = network(test_data)
1414
result = datareader.denormalize3D(np.asarray(result), which='scale')
1515
test_data = datareader.denormalize2D(test_data, which='scale')
1616
test_labels = datareader.denormalize3D(test_labels, which='scale')

tensorlayer/app/computer_vision.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -102,7 +102,7 @@ def __init__(self, model_name='3D-pose'):
102102

103103
def __call__(self, input_data):
104104
if self.model_name == '3D-pose':
105-
output = self.model(input_data)
105+
output = self.model(input_data, is_train=False)
106106
else:
107107
raise NotImplementedError
108108

0 commit comments

Comments
 (0)