Method frameInference

Lib/Model/BasicPitchCNN.cpp:65–95

}

void BasicPitchCNN::frameInference(const float* inData,
                                   std::vector<float>& outContours,
                                   std::vector<float>& outNotes,
                                   std::vector<float>& outOnsets)
{
    // Checks on parameters
    assert(outContours.size() == NUM_FREQ_IN);
    assert(outNotes.size() == NUM_FREQ_OUT);
    assert(outOnsets.size() == NUM_FREQ_OUT);

    // Copy data in aligned input array for inference
    std::copy(inData, inData + NUM_HARMONICS * NUM_FREQ_IN, mInputArray.begin());

    _runModels();

    // Fill output vectors
    std::copy(mCNNOnsetOutput.getOutputs(), mCNNOnsetOutput.getOutputs() + NUM_FREQ_OUT, outOnsets.begin());

    std::copy(mNotesCircularBuffer[(size_t) _wrapIndex(mNoteIdx + 1, mNumNoteStored)].begin(),
              mNotesCircularBuffer[(size_t) _wrapIndex(mNoteIdx + 1, mNumNoteStored)].end(),
              outNotes.begin());

    std::copy(mContoursCircularBuffer[(size_t) _wrapIndex(mContourIdx + 1, mNumContourStored)].begin(),
              mContoursCircularBuffer[(size_t) _wrapIndex(mContourIdx + 1, mNumContourStored)].end(),
              outContours.begin());

    // Increment index for different circular buffers
    mContourIdx = (mContourIdx == mNumContourStored - 1) ? 0 : mContourIdx + 1;
    mNoteIdx = (mNoteIdx == mNumNoteStored - 1) ? 0 : mNoteIdx + 1;
    mConcat2Idx = (mConcat2Idx == mNumConcat2Stored - 1) ? 0 : mConcat2Idx + 1;
}

void BasicPitchCNN::_runModels()
{