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

快来打我* 2022-10-27 05:28 196阅读 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 条评论，196人围观）

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

相关阅读

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

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

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

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

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

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

相关 JAVA深度学习框架DJL之Mnist手写数字0-9识别

1、快速创建项目的插件Alibaba Cloud Toolkit 2、pom添加依赖 <!-- https://mvnrepository.com/artifact

た入场券/ 2023年01月10日 11:48/ 0 赞/ 146 阅读

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

相关 JAVA深度学习框架DJL之COVID19 x-ray图片分类

1、从kaggle 下载数据， ![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmN

分手后的思念是犯贱/ 2022年10月25日 11:22/ 0 赞/ 158 阅读

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

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

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