bug fixed

Former-commit-id: b46106e7862c6c68e5e4065447a46dc00586a834

evaluation/alignment.py

@@ -111,12 +111,12 @@ def alignment(pred_path, gt_path, threshold=0.5):
         gs = gt[idx_t] if idx_t < len(gt) else empt
 
         # Merging sequence for pred
-        while idx_p + 1 < len(pred) and abs(pred[idx_p + 1].end - gs.end) <= threshold:
+        while idx_p + 1 < len(pred) and pred[idx_p + 1].end <= gs.end + threshold:
             ps += pred[idx_p + 1]
             idx_p += 1
 
         # Merging sequence for gt
-        while idx_t + 1 < len(gt) and abs(gt[idx_t + 1].end - ps.end) <= threshold:
+        while idx_t + 1 < len(gt) and gt[idx_t + 1].end <= ps.end + threshold:
             gs += gt[idx_t + 1]
             idx_t += 1
 
@@ -126,8 +126,14 @@ def alignment(pred_path, gt_path, threshold=0.5):
         idx_p += 1
         idx_t += 1
 
+
+    #for a in pred_arr:
+    #    print(a.translation)
+    #for a in gt_arr:
+    #    print(a.source_text)
+
     return zip(pred_arr, gt_arr)
 
 
 #  Test Case
-#alignment('test_translation_zh.srt', 'test_translation_bi.srt')
+#alignment('test_translation_s2.srt', 'test_translation_zh.srt')