Commit d19c6785 authored by sunxx1's avatar sunxx1
Browse files

更新readme

parent db43d34d
[submodule "mmclassification-mmcv"]
path = mmclassification-mmcv
url = http://developer.hpccube.com/codes/aicomponent/mmclassification-mmcv
# Efficientnet_b2
## 论文
EfficientNet: Rethinking Model Scaling for Convolutional Neural Networks
- https://arxiv.org/abs/1905.11946
## 模型介绍
EfficientNet B2是一种卷积神经网络模型,由Google Brain团队于2019年提出。它是EfficientNet系列的一部分,是在ImageNet数据集上进行训练的,具有高度优化的网络结构,可以有效地识别和分类图像。
![20231124102153](./images/20231124102153.png)
## 模型结构
EfficientNet B2模型的网络结构可以分为三个部分:特征提取器、特征增强层和分类器。
首先,输入的图像经过一系列卷积层和池化层进行特征提取。EfficientNet B2使用了轻量级深度可分离卷积层和残差连接来减少计算时间和内存占用,并且采用了一种自适应扩张卷积(SE)模块来进一步提高表示能力。
接下来是特征增强层,它包括了一些额外的层来进一步提高特征表示能力。其中最重要的是MBConv6和MBConv6+两种类型的块。这些块通过复合缩放方法,在每个层面上增加了通道数和分辨率,提高了特征提取器的效率。
![20210419135003777](./images/20210419135003777.png)
最后,将特征向量送入全局平均池化层得到一个固定大小的特征向量,然后通过一组全连接层将其映射到输出类别。分类器部分采用了Dropout正则化来防止过拟合。
## 数据集
在本测试中可以使用ImageNet数据集。
## 环境配置
## Efficientnet_b2训练
### Docker
### 环境配置
```python
git clone --recursive http://developer.hpccube.com/codes/modelzoo/efficientnet_b2_mmcv.git
docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:1.10.0-centos7.6-dtk-22.10.1-py37-latest
# <your IMAGE ID>用以上拉取的docker的镜像ID替换
docker run --shm-size 10g --network=host --name=nit-pytorch --privileged --device=/dev/kfd --device=/dev/dri --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -v $PWD/Efficientnet_b2_mmcv:/home/Efficientnet_b2_mmcv -it <your IMAGE ID> bash
下载工程代码:
cd Densenet121-mmcv/mmclassification-mmcv
pip install -r requirements.txt
```
http://developer.hpccube.com/codes/aicomponent/mmclassification-mmcv.git
## 数据集
提供[光源](https://www.sourcefind.cn/#/service-details)拉取的训练的docker镜像:
在本测试中可以使用ImageNet数据集。
* 训练镜像:docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:1.10.0-centos7.6-dtk-22.10.1-py37-latest
* pip install -r requirements.txt
* cd mmclassification
* pip install -e .
```
├── meta
├── train
├── val
```
### 训练
将训练数据解压到data目录下。
训练命令:
### 单机8卡
./efficientnet.sh
## 准确率数据
## 精度
测试数据使用的是ImageNet数据集,使用的加速卡是DCU Z100L。
......@@ -49,9 +58,23 @@ http://developer.hpccube.com/codes/aicomponent/mmclassification-mmcv.git
| :--: | :-----------------------: |
| 8 | top1:0.73228;top5:0.91522 |
## result
![img](https://developer.hpccube.com/codes/modelzoo/vit_pytorch/-/raw/master/image/README/1695381570003.png)
## 应用场景
### 算法类别
图像分类
### 热点行业
制造,能源,交通,网安
### 源码仓库及问题反馈
https://developer.hpccube.com/codes/modelzoo/efficientnet_b2-mmcv
https://developer.hpccube.com/codes/modelzoo/efficientnet_b2_mmcv
### 参考
......
Subproject commit 0f6a312ab4b30c6e27efd93608268fe0fe3f7dcc
# 模型唯一标识
modelCode = 127
# 模型名称
modelName=Efficientnet_b2-mmcv
modelName=efficientnet_b2_mmcv
# 模型描述
modelDescription=Efficientnet_b2是一种图像分类的卷积神经网络模型,基于MMCV实现测试
# 应用场景
appScenario=训练,图像分类
appScenario=训练,图像分类,制造,能源,交通,网安
# 框架类型
frameType=PyTorch,MMCV
frameType=pytorch,mmcv
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment