AttributeError: 'int' object has no attribute 'item'

[daviduarte]

Erro reproduction:

• Download this video from UCF-Crime dataset:
• Exec:
  python3 scripts/demo.py video.source=Normal_Videos465_x264.mp4 video.output_dir=./output

It will raise an error:

Traceback (most recent call last):
  File "/home/lecun/PHALP/phalp/trackers/PHALP.py", line 205, in track
    self.tracker.update(detections, t_, frame_name, self.cfg.phalp.shot)
  File "/home/lecun/PHALP/phalp/external/deep_sort_/tracker.py", line 83, in update
    self.accumulate_vectors([i[0] for i in matches], features=self.cfg.phalp.predict)
  File "/home/lecun/PHALP/phalp/external/deep_sort_/tracker.py", line 179, in accumulate_vectors
    if("P" in features): p_pred = self.phalp_tracker.forward_for_tracking([p_features, p_data, t_features], "P", l_time)
  File "/home/lecun/PHALP/phalp/trackers/PHALP.py", line 491, in forward_for_tracking
    pose_pred = self.pose_predictor.predict_next(en_pose, en_data, en_time, time)
  File "/home/lecun/PHALP/phalp/models/predictor/pose_transformer_v2.py", line 398, in predict_next
    t_end.append(t.item())
AttributeError: 'int' object has no attribute 'item'

[daviduarte]

Possible solution:

• In PHALP/phalp/models/predictor/pose_transform_v2
• replace
  t_end.append(t.item())
  to
  t_end.append(t.item() if torch.is_tensor(t) else t)