wenet-e2e / wespeaker

Research and Production Oriented Speaker Verification, Recognition and Diarization Toolkit
Apache License 2.0
630 stars 109 forks source link

Feature Request: Use extract_embedding with a batch_size parameter #258

Open asusdisciple opened 7 months ago

asusdisciple commented 7 months ago

It would be nice if you could extract_embeddingwith a batch_size parameter. Also accepting Numpy arrays or torch tensor instead of only file like types would be nice.