単純な畳み込みニューラルネットワークCNNを構築するためのJavaディープラーニングのDJL

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

おすすめ

転載: blog.csdn.net/zsj777/article/details/113481027