xingyizhou/CenterTrack

why does velocity change with aug_s like depth, at least the velocity in x direction?

machengnan opened this issue · 0 comments

if 'velocity' in self.opt.heads:
  if ('velocity' in ann) and min(ann['velocity']) > -1000:
    ret['velocity'][k] = np.array(ann['velocity'], np.float32)[:3]
    ret['velocity_mask'][k] = 1
  gt_det['velocity'].append(ret['velocity'][k])

if 'hps' in self.opt.heads:
  self._add_hps(ret, k, ann, gt_det, trans_output, ct_int, bbox, h, w)

if 'rot' in self.opt.heads:
  self._add_rot(ret, ann, k, gt_det)

if 'dep' in self.opt.heads:
  if 'depth' in ann:
    ret['dep_mask'][k] = 1
    ret['dep'][k] = ann['depth'] * aug_s
    gt_det['dep'].append(ret['dep'][k])
  else:
    gt_det['dep'].append(2)