Skip to content

Sanaxen/trying-using-tiny-dnn

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

16 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

trying using tiny-dnn

Code Examples

input image 32x32x1 channle
tiny-dnn (original)

  nn << conv(32, 32, 5, 1, 6, padding::valid, true, 1, 1, backend_type)
     << tanh()
     << ave_pool(28, 28, 6, 2)
     << tanh()
     << conv(14, 14, 5, 6, 16, connection_table(tbl, 6, 16), padding::valid, true, 1, 1, backend_type)
     << tanh()
     << ave_pool(10, 10, 16, 2)
     << tanh()
     << conv(5, 5, 5, 16, 120, padding::valid, true, 1, 1, backend_type)
     << tanh()
     << fc(120, 10, true, backend_type)
     << tanh();

conv(32, 32, 5, 1, 6...) -> tanh() -> ave_pool(28, 28, 6, 2)
28 = (32 - 5 + 1) / stride
This calculation is troublesome!!

I tried not to calculate the output size

     LayerInfo layers(32, 32, 1);
     nn << layers.add_cnv(6, 5, 1, padding::valid);
     nn << tanh();
     nn << layers.add_avepool(2, 2);
     nn << tanh();
     nn << layers.add_cnv(16, 5, 1, padding::valid, true, connection_table(tbl, 6, 16));
     nn << tanh();
     nn << layers.add_avepool(2, 2);
     nn << tanh();
     nn << layers.add_cnv(120, 5, 1, padding::valid);
     nn << tanh();
     nn << layers.add_fc(10);
     nn << tanh();;

About

No description or website provided.

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published