From 8db0930fd462f2e9e0c65e3ff889558d94fb5df4 Mon Sep 17 00:00:00 2001 From: Paul Brossier Date: Thu, 10 Jan 2019 19:14:31 +0100 Subject: [PATCH] [pitch_crepe] add activations --- src/pitch/pitch_crepe.c | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/src/pitch/pitch_crepe.c b/src/pitch/pitch_crepe.c index f1c7d95a..f9b389df 100644 --- a/src/pitch/pitch_crepe.c +++ b/src/pitch/pitch_crepe.c @@ -36,6 +36,7 @@ #include "fmat.h" #include "ai/tensor.h" +#include "ai/activation.h" #include "ai/conv1d.h" #include "ai/maxpool1d.h" #include "ai/batchnorm.h" @@ -318,6 +319,8 @@ void aubio_pitch_crepe_do(aubio_pitch_crepe_t *o, fvec_t *input, fvec_t *out) for (i = 0; i < o->n_layers; i++) { aubio_conv1d_do(o->conv_layers[i], block_input, o->conv_output[i]); + // relu activation + aubio_activation_relu(o->conv_output[i]); aubio_batchnorm_do(o->batchnorm_layers[i], o->conv_output[i], o->batchnorm_output[i]); aubio_maxpool1d_do(o->maxpool_layers[i], o->batchnorm_output[i], @@ -335,6 +338,9 @@ void aubio_pitch_crepe_do(aubio_pitch_crepe_t *o, fvec_t *input, fvec_t *out) // compute dense layer aubio_dense_do(o->dense_layer, o->flattened, o->dense_output); + // sigmoid activation + aubio_activation_sigmoid(o->dense_output); + #if 0 // print debug output for (i = 0; i < o->n_layers; i++) { -- 2.11.0