-
Notifications
You must be signed in to change notification settings - Fork 117
Open
Description
您好,为什么这里4条车道线都在行列锚上进行了预测,而不是像论文中2条行锚、2条侧锚呢?希望得到您的回复感谢~
pred_dict = {'loc_row': out[:,:self.dim1].view(-1,self.num_grid_row, self.num_cls_row, self.num_lane_on_row),
'loc_col': out[:,self.dim1:self.dim1+self.dim2].view(-1, self.num_grid_col, self.num_cls_col, self.num_lane_on_col),
'exist_row': out[:,self.dim1+self.dim2:self.dim1+self.dim2+self.dim3].view(-1, 2, self.num_cls_row, self.num_lane_on_row),
'exist_col': out[:,-self.dim4:].view(-1, 2, self.num_cls_col, self.num_lane_on_col),
}
Metadata
Metadata
Assignees
Labels
No labels