java深度学习之DJL构建简单的卷积神经网络CNN

1、伪代码

  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())