bug fixed

Former-commit-id: fde89b18c3b8a9bbab04e50cada7f31fd582da86

evaluation/alignment.py

@@ -2,6 +2,7 @@ import sys
 import numpy as np
 sys.path.append('../src')
 from srt_util.srt import SrtScript
+from srt_util.srt import SrtSegment
 
 
 # Helper method
@@ -30,6 +31,7 @@ def procedure(anchor, subsec, S_arr, subidx):
 # Output: aligned array of SRTsegment corresponding to path1 path2
 # Note: Modify comment with .source_text to get output array with string only
 def alignment(pred_path, gt_path):
+    empt = SrtSegment([0,'00:00:00,000 --> 00:00:00,000','','',''])
     pred = SrtScript.parse_from_srt_file(pred_path).segments
     gt = SrtScript.parse_from_srt_file(gt_path).segments
     pred_arr, gt_arr = [], []
@@ -43,14 +45,14 @@ def alignment(pred_path, gt_path):
         if not ps:
             # If ps runs out, align gs segment with filler one by one
             gt_arr.append(gs)#.source_text
-            pred_arr.append('')
+            pred_arr.append(empt)
             idx_t += 1
             continue
 
         if not gs:
             # If gs runs out, align ps segment with filler one by one
             pred_arr.append(ps)#.source_text 
-            gt_arr.append('')
+            gt_arr.append(empt)
             idx_p += 1
             continue
 
@@ -62,7 +64,7 @@ def alignment(pred_path, gt_path):
             # Detect segment with no overlap
             if ps.end < gs.start:
                 pred_arr.append(ps)#.source_text
-                gt_arr.append('')  # append filler
+                gt_arr.append(empt)  # append filler
                 idx_t -= 1  # reset ground truth index
             else:
                 gt_arr.append(gs)#.source_text
@@ -70,13 +72,13 @@ def alignment(pred_path, gt_path):
                     pred_arr.append(ps)#.source_text
                     idx_p = procedure(gs, pred, pred_arr, idx_p + 1)
                 else:  # filler pairing
-                    pred_arr.append('')
+                    pred_arr.append(empt)
                     idx_p -= 1
         else:
             # same overlap checking procedure
             if gs.end < ps.start:
                 gt_arr.append(gs)#.source_text
-                pred_arr.append('')  # filler
+                pred_arr.append(empt)  # filler
                 idx_p -= 1  # reset
             else:
                 pred_arr.append(ps)#.source_text
@@ -84,13 +86,14 @@ def alignment(pred_path, gt_path):
                     gt_arr.append(gs)#.source_text
                     idx_t = procedure(ps, gt, gt_arr, idx_t + 1)
                 else:  # filler pairing
-                    gt_arr.append('')
+                    gt_arr.append(empt)
                     idx_t -= 1
 
         idx_p += 1
         idx_t += 1
-    #print(gt_arr)
+    #for a in pred_arr:
+    #    print(a.source_text)
     return zip(pred_arr, gt_arr)
 
 #  Test Case
-#alignment('../results/...PATH1.../FILE.srt', '../results/PATH2/FILE.srt')
+#alignment('test_translation_zh.srt', 'test_translation_bi.srt')