search

beibuwandeluori / DFGC_Detection

The 1st solution for the DeepFake Game Competition (DFGC) @ IJCB 2021 Detection track.
33 stars 2 forks source link

IndexError: index 0 is out of bounds for axis 0 with size 0 #3

Open ningerlove opened 2 years ago

ningerlove commented 2 years ago

File "extract_video_celeb_df_v2.py", line 168, in main() File "extract_video_celeb_df_v2.py", line 132, in main face_images, face_boxes, face_index = extract_video(real_video_path, model, scale=scale) File "extract_video_celeb_df_v2.py", line 57, in extract_video method="probability") File "/usr/local/lib/python3.6/dist-packages/facenet_pytorch/models/mtcnn.py", line 436, in select_boxes box = boxes[box_order][[0]] IndexError: index 0 is out of bounds for axis 0 with size 0 how to solve?

samils7 commented 2 years ago

Make sure you have the exact version of facenet-pytorch==2.5.0 It will solve the problem