gorjanradevski / vsepp_tensorflow Goto Github PK

Hi, I'm working on text-image retrieval and I appreciate your tensorflow version of vesepp model.

During the code review, I'm confused about the implementation of the function text2image_recall_at_k in evaluator.py. The original code is shown as below:

While I think that line 117 should be like this, since the embedded_captions cannot be sliced 5 fold:

for index in range(num_captions//5):

Interestingly, this change does not change the final evaluation result.