mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-12-11 06:55:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

from local

Browse Source
This commit is contained in:
dohe0342 2022-12-10 12:49:07 +09:00
parent bd9c10f9f1
commit 552f89adee
2 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

BIN
icefall/.utils.py.swp
View File

Binary file not shown.

8
icefall/utils.py
View File

@ -191,7 +191,15 @@ def encode_supervisions(
returned tensor and list of strings are guaranteed to be consistent with
each other.
"""
try: start_frame = supervisions["start_frame"]
except: start_frame = [0 for i in range(len(supervisions["cut"]))]
try: num_frames = supervisions["num_frames"]
except:
num_frames = []
for supervision in supervisions['cut']:
try: num_frames.append(supervision.tracks[0].cut.recording.num_samples)
except: num_frames.append(supervision.recording.num_samples)
supervision_segments = torch.stack(
(
supervisions["sequence_idx"],