AlexNet模型实现（1. C++宏定义头文件）

最新推荐文章于 2023-11-20 22:02:13 发布

aitazhixin

最新推荐文章于 2023-11-20 22:02:13 发布

阅读量2k

点赞数

分类专栏： AI 算法文章标签： AlexNet CNN

本文链接：https://blog.csdn.net/aitazhixin/article/details/73322145

版权

定义了模型架构需要的参数和激活函数：

// AlexNet define
#ifndef __SLQ_NET_MACRO_H__
#define __SLQ_NET_MACRO_H__

#include <cmath>
namespace slqDL {
  namespace slqAlexNet{
    
#define EpochLoop           (100)
#define AccuracyRate        (0.56)
#define Alpha               (0.01f)
#define LoopError           (0.001f)
#define EspCNN              (1e-8f)
    
#define poolSpace                    (3)
#define poolStride                   (2)

#define TrainImgNum                  (40000)
#define TestImgNum                   (10000)
    
#define inMapHigh                    (227)
#define inMapWidth                   (227)
#define inMapNum                     (3)
#define inMapSize                    (51529)                         // 227*227
#define inUnitNum                    (154587)                        // 227*227*3
    
#define c1MapHigh                    (55)
#define c1MapWidth                   (55)
#define c1MapNum                     (96)
#define c1MapSize                    (3025)                          // 55*55
#define c1UnitNum                    (290400)                        // 55*55*96
    
#define c1ConvHigh                   (11)
#define c1ConvWidth                  (11)
#define c1ConvDeep                   (3)
#define c1ConvStride                 (4)
#define c1ConvTensor                 (121)