search

salaniz / pytorch-gve-lrcn

PyTorch implementations for "Generating Visual Explanations" (GVE) and "Long-term Recurrent Convolutional Networks" (LRCN)
MIT License
92 stars 22 forks source link

Class predictions not outputted in results dictionary #21

Closed katelyn98 closed 1 year ago

katelyn98 commented 1 year ago

I am hoping to not only see the generated sentence, but also the predicted class from the image model. I have been searching through the code to find where the class prediction is, but I cannot find it anywhere. Can someone please help me locate in the generate_sentence or eval functions where the model is predicting the image class so I can output the predicted class as well as the generated sentence?

katelyn98 commented 1 year ago

I found the predictions from the bilinear model in one of the pickle files. Thanks.