fixed estimator import in both ros nodes

Prepended package name to estimator in order to resolve import failures.
bvanelli · Jun 1, 2018 · 9d2e370 · 9d2e370
1 parent 07ba613
commit 9d2e370
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/scripts/broadcaster_ros.py b/scripts/broadcaster_ros.py
@@ -12,8 +12,8 @@
 from sensor_msgs.msg import Image
 from tfpose_ros.msg import Persons, Person, BodyPartElm
 
-from estimator import TfPoseEstimator
-from networks import model_wh, get_graph_path
+from tf_pose.estimator import TfPoseEstimator
+from tf_pose.networks import model_wh, get_graph_path
 
 
 def humans_to_msg(humans):

diff --git a/scripts/visualization.py b/scripts/visualization.py
@@ -6,7 +6,7 @@
 from cv_bridge import CvBridge, CvBridgeError
 
 from tfpose_ros.msg import Persons, Person, BodyPartElm
-from estimator import Human, BodyPart, TfPoseEstimator
+from tf_pose.estimator import Human, BodyPart, TfPoseEstimator
 
 
 class VideoFrames: