search

bill9800 / speech_separation

Include some core functions and model to handle speech separation
MIT License
153 stars 61 forks source link

IndexError in AV_model_eval.py at parse_X_data() #26

Open DunkinDonat opened 3 years ago

DunkinDonat commented 3 years ago

Hi there, wehen i am trying to predict there is this error going on:

line 41 in speech_separation/model/model_v2/AV_model_eval.py /

face_embs[1, :, :, :, i] = np.load(face_path + single_idxs[i] + "_face_emb.npy")

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

face_embs shape is (1,75,1,1972,2) i can be in my case 0 or 1 np.load(face_path + single_idxs[i] + "_face_emb.npy") shape is (75,1,1972)

Whats wrong here? Do we need to change line 41 from face_embs[1, :, :, :, i] to from face_embs[0, :, :, :, i]

Greetings:)

SutirthaChakraborty commented 2 years ago

Did you solve it ? @DunkinDonat