BUG: Pose processing, open cv error: assertion failed

[niksart]

Version information:

facemap version: 1.0.6
platform: win32
python version: 3.8.19
torch version: 2.4.1+cpu
numpy version: 1.24.4

Describe the issue:

Hello, I get this error related to OpenCV at the end of the pose processing. The pose processing finishes (the bar arrives almost at the end), but no .h5 file is produced. Thank you in advance for your help!

Error message:

Traceback (most recent call last):
  File "C:\Users\aburgalossi\anaconda3\envs\new_facemap\lib\site-packages\facemap\gui\gui.py", line 1298, in process_ROIs
    self.pose_model.run()
  File "C:\Users\aburgalossi\anaconda3\envs\new_facemap\lib\site-packages\facemap\pose\pose.py", line 457, in run
    pred_data, metadata = self.predict_landmarks(video_id)
  File "C:\Users\aburgalossi\anaconda3\envs\new_facemap\lib\site-packages\facemap\pose\pose.py", line 288, in predict_landmarks
    imall = utils.get_batch_frames(
  File "C:\Users\aburgalossi\anaconda3\envs\new_facemap\lib\site-packages\facemap\utils.py", line 448, in get_batch_frames
    frame = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)[np.newaxis, ...]
cv2.error: OpenCV(4.10.0) D:\a\opencv-python\opencv-python\opencv\modules\imgproc\src\color.cpp:196: error: (-215:Assertion failed) !_src.empty() in function 'cv::cvtColor'

[Atika-Syeda]

Hi @niksart can you please provide more information about the video filetype loaded and if you tried single video or batch processing?