Closed abhiray92 closed 3 years ago
I cannot comment on what's going on as your log is partial.
One thing I can think of is if you are NOT testing over all the video samples (instead you test part of them), you have to make sure at least one video per class is present. Otherwise, you will get NAN for all the classes with no testing videos.
@dxli94 I trained my model on 100 glosses and I used the same nslt_100.json file for testing. Yet it still gives me NaN. Moreover, if I want to check for just 1 video (.mp4 file) in particular, it wouldn't process that either.
The testing code should work as expected, as verified by other users. I could not reproduce the issue, either. Instead, I suggest you set checkpoints and step into the code, in order to understand what is going on.
@dxli94 Thanks, I believe I didn't understand the code well. However, can you please let me know how to use the function 'load_rgb_frames_from_video'? Reason being, I would like to predict the label for a specific video. I don't think the function is being used if i provide the following arguments -
mode = 'rgb'
num_classes = 100
save_model = '/content/drive/MyDrive/I3D_WLASL/code/checkpoints/'
root = '/content/drive/MyDrive/I3D_WLASL/code/data/WLASL2000/'
train_split = '/content/drive/MyDrive/I3D_WLASL/code/preprocess/nslt_{}.json'.format(num_classes)
weights = '/content/drive/MyDrive/I3D_WLASL/code/archived/asl100/FINAL_nslt_100_iters=896_top1=65.89_top5=84.11_top10=89.92.pt'
run(mode=mode, root=root, save_model=save_model, train_split=train_split, weights=weights)
Any help would be highly appreciated! Thanks alot!
Hi there, I was trying to test the trained model, but I am getting the below error while testing -