example_conv2dlayer_predict.cpp

 
#include <Pds/Ra>
#include <Pds/Nn>
 
int main(void)
{
    Pds::SampleBlock Out;
    unsigned int L=5;
    
    // Input file
    Pds::SampleBlock In=Pds::Matrix::ImportBmpFile("../manual/imagenes/pimienta-gray3.bmp");
    //Pds::Matrix::ExportBmpFile(In[0],In[1],In[2],"in.bmp");
    
    // Filtro Gabor
    Pds::Matrix Gabor=Pds::Kernel2D::Gabor(L,6.0,5.5,Pds::Ra::Pi/4,8.0,0.0)/3;
    Pds::Octave::Plot::ImageSc(Gabor,"testando.m","gabor5.png");
    
    // Filtro Mean
    Pds::Matrix Mean=Pds::Kernel2D::Mean(L)/3;
    
    // Tensor con 2 filtros de 3 capas.
    Pds::TensorBlock W={{Gabor,Gabor,Gabor},{Mean,Mean,Mean}};
    
    Pds::Conv2DLayer Conv(W,Pds::Identity,Pds::DIdentity,{1, 1},Pds::Ra::Same);
    
    Out=Conv.Predict(In);
    
    for(unsigned int n=0;n<Out.size();n++)
        Out[n].ExportBmpFile(Pds::Colormap::Gray,"out"+std::to_string(n)+".bmp");
    
    return 0;
}