Update app.py
Browse files
app.py
CHANGED
@@ -202,8 +202,8 @@ def Visual_re_ranker(caption_man, caption_woman, context_label, context_prob):
|
|
202 |
sim_w = get_sim(sim_w)
|
203 |
|
204 |
|
205 |
-
LM_man =
|
206 |
-
LM_woman =
|
207 |
#LM = scorer.sentence_score(caption, reduce="mean")
|
208 |
score_man = pow(float(LM_man),pow((1-float(sim_m))/(1+ float(sim_m)),1-float(context_prob)))
|
209 |
score_woman = pow(float(LM_woman),pow((1-float(sim_w))/(1+ float(sim_w)),1-float(context_prob)))
|
|
|
202 |
sim_w = get_sim(sim_w)
|
203 |
|
204 |
|
205 |
+
LM_man = sentence_prob_mean(caption_man)
|
206 |
+
LM_woman = sentence_prob_mean(caption_woman)
|
207 |
#LM = scorer.sentence_score(caption, reduce="mean")
|
208 |
score_man = pow(float(LM_man),pow((1-float(sim_m))/(1+ float(sim_m)),1-float(context_prob)))
|
209 |
score_woman = pow(float(LM_woman),pow((1-float(sim_w))/(1+ float(sim_w)),1-float(context_prob)))
|