qihfang commited on
Commit
c9b39f1
·
1 Parent(s): 5f8f9fa

Fix AngleChangeEvaluator

Browse files
Files changed (1) hide show
  1. estimator.py +1 -1
estimator.py CHANGED
@@ -227,7 +227,7 @@ class AngleChangeEvaluator(BaseEvaluator):
227
  cos_angle = (ba * bc).sum().clamp(-1.0, 1.0)
228
  return torch.acos(cos_angle) * 180.0 / torch.pi
229
 
230
- angle_diff = torch.abs(compute_angle(self.frame2) - compute_angle(self.frame1))
231
  return torch.tensor([angle_diff > self.threshold]) if self.greater_than else torch.tensor([angle_diff < self.threshold])
232
 
233
 
 
227
  cos_angle = (ba * bc).sum().clamp(-1.0, 1.0)
228
  return torch.acos(cos_angle) * 180.0 / torch.pi
229
 
230
+ angle_diff = torch.abs(compute_angle(-1) - compute_angle(0))
231
  return torch.tensor([angle_diff > self.threshold]) if self.greater_than else torch.tensor([angle_diff < self.threshold])
232
 
233