Commit 85f1d0d5 authored by Cedarch's avatar Cedarch
Browse files

update yaml of MPPNet

parent 4a8a3bf7
......@@ -225,10 +225,7 @@ class MPPNetHeadE2E(RoIHeadTemplate):
time_stamp[:,i,:] = i*0.1
box_seq = torch.cat([trajectory_rois[:,:,:,:7],time_stamp],-1)
if self.model_cfg.USE_BOX_ENCODING.NORM_T0:
# canonical transformation
box_seq[:, :, :,0:3] = box_seq[:, :, :,0:3] - box_seq[:, 0:1, :, 0:3]
box_seq[:, :, :,0:3] = box_seq[:, :, :,0:3] - box_seq[:, 0:1, :, 0:3]
roi_ry = box_seq[:,:,:,6] % (2 * np.pi)
......
......@@ -91,13 +91,12 @@ MODEL:
ENABLED: True
NORM_T0: True
ALL_YAW_T0: True
AVG_STAGE_1: True
AVG_STAGE1_SCORE: True
USE_TRAJ_EMPTY_MASK: True
USE_AUX_LOSS: True
USE_MLP_JOINTEMB: True
IOU_WEIGHT: [0.5,0.4]
ROI_GRID_POOL: #if using 16frame, change to corresponding setting
ROI_GRID_POOL: #if using 16frame, change to the corresponding setting
GRID_SIZE: 4
MLPS: [[128,128], [128,128]]
POOL_RADIUS: [0.8, 1.6]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment