コード例 #1
0
ファイル: demo.py プロジェクト: nwnuzh/LipNet-PyTorch
def ctc_decode(y):
    y = y.argmax(-1)
    t = y.size(0)
    result = []
    for i in range(t+1):
        result.append(MyDataset.ctc_arr2txt(y[:i], start=1))
    return result
コード例 #2
0
def ctc_decode(y):
    result = []
    y = y.argmax(-1)
    return [MyDataset.ctc_arr2txt(y[_], start=1) for _ in range(y.size(0))]
コード例 #3
0
ファイル: main.py プロジェクト: blueCat2020/LipNet
def ctc_decode(y):
    y = y.argmax(-1)
    return [MyDataset.ctc_arr2txt(y[_]) for _ in range(y.size(0))]