java深度学习之DJL创建NiN网络

快来打我* 2022-10-27 05:28 207阅读 0赞

1、NiN块由一个卷积层和两个1×1卷积层组成。

构建NiN块代码如下

public static SequentialBlock niNBlock(int numChannels, Shape kernelShape,
                                               Shape strideShape, Shape paddingShape) {
    
            SequentialBlock tempBlock = new SequentialBlock();
            // numChannels  通道数 滤波器层数
            //kernelShape 卷积核大小
            //strideShape 步幅
            // paddingShape  填充大小
            tempBlock.add(Conv2d.builder()
                    .setKernelShape(kernelShape)
                    .optStride(strideShape)
                    .optPadding(paddingShape)
                    .setFilters(numChannels)
                    .build())
                    .add(Activation::relu)
                    .add(Conv2d.builder()
                            .setKernelShape(new Shape(1, 1))
                            .setFilters(numChannels)
                            .build())
                    .add(Activation::relu)
                    .add(Conv2d.builder()
                            .setKernelShape(new

发表评论取消回复

表情：

评论列表（有 0 条评论，207人围观）

还没有评论，来说两句吧...

相关阅读

相关 Java深度学习库DJL实现Python的NumPy

目录 1 NDArray 的背景介绍 1.1 架构 2 JavaDJL使用 2.1 安装DJL 2.2 基本操作

秒速五厘米/ 2024年04月07日 12:54/ 0 赞/ 80 阅读

相关 djl和ljl_使用Spring Boot和DJL进行深度学习

djl和ljl 总览这是Spring Boot上的另一篇[文章][Link 1] ，该文章将展示如何使用[Deep Java Library][] （DJL）构建示例

矫情吗；*/ 2023年02月25日 03:56/ 0 赞/ 90 阅读

相关 JAVA深度学习框架DJL之鞋子分类

机器学习生命周期遵循机器学习生命周期来生成鞋类分类模型。 ML生命周期不同于传统的软件开发生命周期，它包含六个具体步骤： 1. 获取数据 2. 清理并准备数据 3

红太狼/ 2023年01月11日 11:00/ 0 赞/ 283 阅读

相关 java深度学习之DJL卷积神经网络CNN

伪代码代码使用3层卷积神经网络conv 和两层全连接Fully Connected Layer // conv -> conv -> conv -> fc ->

小鱼儿/ 2023年01月11日 03:49/ 0 赞/ 200 阅读

相关 java深度学习之DJL创建DenseNet网络

package com.example.demo.djl; import ai.djl.Model; import ai.djl.ModelE

怼烎@/ 2022年10月28日 14:26/ 0 赞/ 293 阅读

相关 java深度学习之DJL创建AlexNet

package com.example.demo.djl; import ai.djl.Model; import ai.djl.ModelE

ゞ浴缸里的玫瑰/ 2022年10月27日 15:49/ 0 赞/ 175 阅读

相关 java深度学习之DJL创建NiN网络

1、NiN块由一个卷积层和两个1×1卷积层组成。构建NiN块代码如下 public static SequentialBlock niNBlock(int num

快来打我*/ 2022年10月27日 05:28/ 0 赞/ 208 阅读

相关 java深度学习之DJL迁移学习

1、数据集很小，数据相似性很高考虑这个kaggle数据集(https://www.kaggle.com/mriganksingh/cat-images-dataset)。这

迈不过友情╰/ 2022年10月26日 04:57/ 0 赞/ 217 阅读

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

1、伪代码 Model model = Model.newInstance(MODEL_NAME); // Block is a c

蔚落/ 2022年10月25日 11:29/ 0 赞/ 229 阅读

相关深度学习经典网络，AlexNet,NIN,OverFeat,VGG,ResNet

深度学习经典网络 1.AlexNet \[2012\] 2.Network In Network \[2013\]

分手后的思念是犯贱/ 2022年05月10日 02:24/ 0 赞/ 376 阅读