Problem during inference

[Utkarsh-quytech]

While running the demo.py on wideresnet_mars.pth or siamesenet_mars.pth the terminal gets stuck unless i close the video window, then % increases and then after some windows are closed it starts giving the below error. Any help is much appreciated, Thanks!!

(humanaction) C:\Users\Utkarsh.Singh\Documents\Utkarsh\PoseTracking\human-action-recognition\src>python demo.py --task action --source ../test_data/fun_theory.mp4 --save_folder ../output --debug_track pycuda or tensorrt not installed. [INFO] Loading pytorch trtpose model with "../weights\pose_estimation\trtpose\densenet121_baseline_att_256x256_B_epoch_160.pth" [INFO] Loading pytorch reid model: ../weights\tracker\deepsort\wideresnet_mars.pth. [INFO] Writing output to ../output\fun_theory_trtpose_deepsort_mars_wideresnet_dnn_torch.avi [INFO] Writing output to ../output\fun_theory_trtpose_deepsort_mars_wideresnet_dnn_torch_debug.avi Run | fun_theory.mp4 ---------------------------------------- 0% -:--:-- 66.85fps[INFO] Saving video to : ../output\fun_theory_trtpose_deepsort_mars_wideresnet_dnn_torch.avi Run | fun_theory.mp4 ---- ----------------------------------- 11% -:--:-- 0.03fps Traceback (most recent call last): File "demo.py", line 138, in main() File "demo.py", line 98, in main predictions = action_classifier.classify(predictions) File "C:\Users\Utkarsh.Singh\Documents\Utkarsh\PoseTracking\human-action-recognition\src\lib\classifier\dnn\classifier.py", line 70, in classify self.dict_id2clf[id] = self._create_classifier(id) File "C:\Users\Utkarsh.Singh\Documents\Utkarsh\PoseTracking\human-action-recognition\src\lib\classifier\dnn\classifier.py", line 53, in model_path, classes, window_size, human_id, threshold=threshold) File "C:\Users\Utkarsh.Singh\Documents\Utkarsh\PoseTracking\human-action-recognition\src\lib\classifier\dnn\classifier.py", line 172, in init self.model = pickle.load(f) ModuleNotFoundError: No module named 'sklearn.neighbors._classification'

[Raven19888]

I'm having the same problem, which seems to be due to an incompatibility with the scikit-learn version, but I haven't found a solution yet