example_customcnn_create.cpp

 
#include <Pds/Ra>
#include <Pds/Nn>
 
Pds::CustomCNn GenerateModel(   unsigned int Nch,
                                unsigned int Nlin,
                                unsigned int Ncol)
{
    Pds::MaxPoolingLayer Max(2,2);
    Pds::FlattenLayer Flat;
    
    Pds::CustomCNn Model(Nch,Nlin,Ncol);
    
    Pds::Conv2DLayer Conv1(3,16,Pds::Size(3,3),Pds::ReLU, Pds::DReLU,{1, 1},Pds::Ra::Valid);
    Model.Add(Conv1.Clone());
    
    Model.Add(Max.Clone());
    
    Pds::Conv2DLayer Conv2(16, 8,Pds::Size(3,3),Pds::ReLU, Pds::DReLU,{1, 1},Pds::Ra::Valid);
    Model.Add(Conv2.Clone());
    
    Model.Add(Max.Clone());
        
    Pds::Conv2DLayer Conv3(8, 4,Pds::Size(3,3),Pds::ReLU, Pds::DReLU,{1, 1},Pds::Ra::Valid);
    Model.Add(Conv3.Clone());
    
    Model.Add(Max.Clone());
        
    Pds::Conv2DLayer Conv4(4, 2,Pds::Size(3,3),Pds::Tanh, Pds::DTanh,{1, 1},Pds::Ra::Valid);
    Model.Add(Conv4.Clone());
    
    Model.Add(Flat.Clone());
 
    Pds::FCLayer FC1(2048,2048,Pds::Tanh,Pds::DTanh);
    Model.Add(FC1.Clone());
    
    Pds::FCLayer FC2(2048,512,Pds::Tanh,Pds::DTanh);
    Model.Add(FC2.Clone());
    
    Pds::FCLayer FC3(512,3,Pds::Sigmoid,Pds::DSigmoid);
    Model.Add(FC3.Clone());
    
    return Model;
}
 
 
int main(void)
{
    unsigned int Nch=3;
    unsigned int Nlin=284;
    unsigned int Ncol=Nlin;
    
    Pds::CustomCNn Model=GenerateModel(Nch,Nlin,Ncol);
    Model.Print("Model:\n"); 
    
    Pds::SampleBlock Img=Pds::Matrix::ImportBmpFile("../manual/imagenes/pimienta284.bmp");
    
    Pds::TensorBlock Block={Img};
    
    Block=Model.PredictTensor(Block);
    
    for(unsigned int n=0;n<Block.size();n++)
    for(unsigned int m=0;m<Block[n].size();m++)
        Block[n][m].Print("\nBlock["+std::to_string(n)+"]["+std::to_string(m)+"]:\n");
    
    
    std::ofstream myfile("Model.xml", std::ofstream::out);
    Model.ExportXmlToStream(myfile);
    myfile.close();
    
    return 0;
}