雾聪
2024-01-03 fcaf910ceb4e07a13bf2d133f46df684b069b3f0
runtime/onnxruntime/src/util.cpp
@@ -305,6 +305,10 @@
}
bool TimestampIsPunctuation(U16CHAR_T &u16) {
    // (& ' -) in the dict
    if (u16 == 0x26 || u16 == 0x27 || u16 == 0x2D){
        return false;
    }
    return (u16 >= 0x21 && u16 <= 0x2F)     // 标准ASCII标点
        || (u16 >= 0x3A && u16 <= 0x40)     // 标准ASCII标点
        || (u16 >= 0x5B && u16 <= 0x60)     // 标准ASCII标点
@@ -580,7 +584,8 @@
                }
            }
            // format
            ts_sent += "{'text':'" + text_seg + "',";
            ts_sent += "{'text_seg':'" + text_seg + "',";
            ts_sent += "'punc':'" + characters[idx_str] + "',";
            ts_sent += "'start':'" + to_string(start) + "',";
            ts_sent += "'end':'" + to_string(end) + "',";
            ts_sent += "'ts_list':" + VectorToString(ts_seg) + "}";
@@ -590,9 +595,7 @@
            } else{
                ts_sentences += ts_sent + ",";
            }
            // clear
            idx_str++;
            text_seg = "";
            ts_sent = "";
            start = 0;
@@ -605,9 +608,9 @@
                text_seg += " " + characters[idx_str];
            }
            ts_seg.push_back(timestamps[idx_ts]);
            idx_str++;
            idx_ts++;
        }
        idx_str++;
    }
    // for none punc results
    if(ts_seg.size() >0){
@@ -618,7 +621,8 @@
            end = ts_seg[ts_seg.size()-1][1];
        }
        // format
        ts_sent += "{'text':'" + text_seg + "',";
        ts_sent += "{'text_seg':'" + text_seg + "',";
        ts_sent += "'punc':'',";
        ts_sent += "'start':'" + to_string(start) + "',";
        ts_sent += "'end':'" + to_string(end) + "',";
        ts_sent += "'ts_list':" + VectorToString(ts_seg) + "}";