diff --git a/mushr_rhc_ros/launch/sim/sim_server_eval.launch b/mushr_rhc_ros/launch/sim/sim_server_eval.launch index b9f5dd8..d4eb742 100644 --- a/mushr_rhc_ros/launch/sim/sim_server_eval.launch +++ b/mushr_rhc_ros/launch/sim/sim_server_eval.launch @@ -10,18 +10,23 @@ - - + + + - - + + + + + + - + diff --git a/mushr_rhc_ros/src/rhcnode_network_pcl_new.py b/mushr_rhc_ros/src/rhcnode_network_pcl_new.py index f3ec23e..af52bb5 100755 --- a/mushr_rhc_ros/src/rhcnode_network_pcl_new.py +++ b/mushr_rhc_ros/src/rhcnode_network_pcl_new.py @@ -168,7 +168,7 @@ def __init__(self, dtype, params, logger, name): map_decoder='deconv', map_recon_dim=64, freeze_core=False, state_loss_weight=0.1, loc_x_loss_weight=0.01, loc_y_loss_weight=0.1, loc_angle_loss_weight=10.0, - loc_decoder_type='separate') + loc_decoder_type='joint') map_model = GPT(mconf_map, device) # map_model=torch.nn.DataParallel(map_model) @@ -185,6 +185,7 @@ def __init__(self, dtype, params, logger, name): map_model.eval() map_model.to(device) self.map_model = map_model + rate_map_display = 1.0 self.map_viz_timer = rospy.Timer(rospy.Duration(1.0 / rate_map_display), self.map_viz_cb) # localization model @@ -216,6 +217,7 @@ def __init__(self, dtype, params, logger, name): loc_model.eval() loc_model.to(device) self.loc_model = loc_model + rate_loc_display = 20 self.map_viz_loc = rospy.Timer(rospy.Duration(1.0 / rate_loc_display), self.loc_viz_cb) @@ -236,9 +238,8 @@ def __init__(self, dtype, params, logger, name): self.time_so_far = 0.0 self.file_name = os.path.join(self.out_path,'info.csv') - # set timer callbacks for visualization - rate_map_display = 1.0 - rate_loc_display = 20 + +