Skip to content

Commit

Permalink
Rename _processor with _preProcessor (#17)
Browse files Browse the repository at this point in the history
  • Loading branch information
kaiidams authored Oct 5, 2023
1 parent de85f4e commit 3f36715
Show file tree
Hide file tree
Showing 2 changed files with 8 additions and 8 deletions.
8 changes: 4 additions & 4 deletions NeMoOnnxSharp/EncDecCTCModel.cs
Original file line number Diff line number Diff line change
Expand Up @@ -16,17 +16,17 @@ public class EncDecCTCModel : ASRModel, IDisposable
{
private const string Vocabulary = " abcdefghijklmnopqrstuvwxyz'_";

private readonly IAudioPreprocessor<short, float> _processor;
private readonly IAudioPreprocessor<short, float> _preProcessor;
private readonly CharTokenizer _tokenizer;
private readonly InferenceSession _inferSess;
private readonly int _features;

public IAudioPreprocessor<short, float> Processor => _processor;
public IAudioPreprocessor<short, float> PreProcessor => _preProcessor;

private EncDecCTCModel(InferenceSession inferSess)
{
_features = 64;
_processor = new AudioToMelSpectrogramPreprocessor(
_preProcessor = new AudioToMelSpectrogramPreprocessor(
sampleRate: 16000,
window: WindowFunction.Hann,
windowSize: 0.02,
Expand Down Expand Up @@ -55,7 +55,7 @@ public void Dispose()
public override string Transcribe(Span<short> inputSignal)
{
string text = string.Empty;
var processedSignal = _processor.GetFeatures(inputSignal);
var processedSignal = _preProcessor.GetFeatures(inputSignal);
processedSignal = TransposeInputSignal(processedSignal, _features);
var container = new List<NamedOnnxValue>();
var audioSignalData = new DenseTensor<float>(
Expand Down
8 changes: 4 additions & 4 deletions NeMoOnnxSharp/EncDecClassificationModel.cs
Original file line number Diff line number Diff line change
Expand Up @@ -28,17 +28,17 @@ public class EncDecClassificationModel : ASRModel, IDisposable
"speech"
};

private readonly IAudioPreprocessor<short, float> _processor;
private readonly IAudioPreprocessor<short, float> _preProcessor;
private readonly InferenceSession _inferSess;
private readonly int _nMelBands;
private readonly string[] _labels;

public IAudioPreprocessor<short, float> Processor => _processor;
public IAudioPreprocessor<short, float> PreProcessor => _preProcessor;

private EncDecClassificationModel(InferenceSession inferSess, bool speechCommands)
{
_nMelBands = 64;
_processor = new AudioToMFCCPreprocessor(
_preProcessor = new AudioToMFCCPreprocessor(
sampleRate: 16000,
window: WindowFunction.Hann,
windowSize: 0.025,
Expand Down Expand Up @@ -69,7 +69,7 @@ public void Dispose()
public override string Transcribe(Span<short> inputSignal)
{
string text = string.Empty;
var processedSignal = _processor.GetFeatures(inputSignal);
var processedSignal = _preProcessor.GetFeatures(inputSignal);
processedSignal = TransposeInputSignal(processedSignal, _nMelBands);
var container = new List<NamedOnnxValue>();
var audioSignalData = new DenseTensor<float>(
Expand Down

0 comments on commit 3f36715

Please sign in to comment.