[src] Fix rare case when segment end rounding overshoots file end in …

…extract-segments (#3331)
kaldi-asr · May 17, 2019 · 528e072 · 528e072
1 parent c8b93bc
commit 528e072
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/featbin/extract-segments.cc b/src/featbin/extract-segments.cc
@@ -180,7 +180,10 @@ int main(int argc, char *argv[]) {
       // conversion requires a proper rounding.
       int32 start_samp = static_cast<int32>(start * samp_freq + 0.5f),
           end_samp = static_cast<int32>(end * samp_freq + 0.5f);
-
+
+      if (end_samp > num_samp) 
+        end_samp = num_samp;
+
       // Get the range of data from the orignial wave_data matrix.
       SubMatrix<BaseFloat> segment_matrix(wave_data, channel, 1,
                                           start_samp, end_samp - start_samp);