search

DGU-PoliceLab / System_Integration

This is PoliceLab 2.0 AI module repository.
GNU General Public License v3.0
1 stars 0 forks source link

MHNCITY Violence 모듈에서 오류 발생 #8

Open qqaazz0222 opened 3 months ago

qqaazz0222 commented 3 months ago

evaluate_frames 함수 동작시 오류 발생

Process Process-4:
Traceback (most recent call last):
  File "/opt/conda/lib/python3.10/multiprocessing/process.py", line 314, in _bootstrap
    self.run()
  File "/opt/conda/lib/python3.10/multiprocessing/process.py", line 108, in run
    self._target(*self._args, **self._kwargs)
  File "/System_Integration/_HAR/MHNCITY/violence.py", line 76, in Violence
    max_score_1, mean_score_1  = evaluate_frames(fight_model_1, skeletons)
  File "/System_Integration/_HAR/MHNCITY/model.py", line 202, in evaluate_frames
    all_scores = model.forward(all_keypoint_batches)
  File "/System_Integration/_HAR/MHNCITY/model.py", line 100, in forward
    flattened_tensor = batch_keypoints_tensor.view(-1, 2)
RuntimeError: shape '[-1, 2]' is invalid for input of size 3