huichen98

去噪任务的影响

6

# noise on the box if self.dn_bbox_noise_scale > 0: wlh = known_bbox_expand[..., 3:6].clone() rand_prob = torch.rand_like(known_bbox_expand) * 2 - 1.0 known_bbox_expand[..., 0:3] += torch.mul(rand_prob[..., 0:3], wlh / 2) * self.dn_bbox_noise_scale...