DJL de Java Deep Learning para construir una red neuronal convolucional simple CNN

1. Pseudocódigo

  Model model = Model.newInstance(MODEL_NAME);

        // Block is a composable unit that forms a neural network; combine them like Lego blocks
        // to form a complex network
        Block block = new SequentialBlock()
                .add(Conv2d.builder()
                        //卷积核
                        .setKernelShape(new Shape(5, 5))
                        //步幅
                        .optStride(new Shape(5, 5))
                        //padding
                        .optPadding(new Shape(2, 2))
                        //过滤器
                        .setFilters(3)
                        .build())
                .add(Activation::relu)

                .add(Conv2d.builder()
                        .setKernelShape(new Shape(5, 5))
                        .optStride(new Shape(3, 3))
                        .optPadding(new Shape(1, 1))
                        .setFilters(3)
                        .build())
          

Supongo que te gusta

Origin blog.csdn.net/zsj777/article/details/113481027
Recomendado
Clasificación