Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
ModelZoo
MobileNetV2_mmcv
Commits
f4e62d75
Commit
f4e62d75
authored
Dec 08, 2023
by
dcuai
Browse files
Update README.md
parent
8e1d9fd8
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
19 additions
and
13 deletions
+19
-13
README.md
README.md
+19
-13
No files found.
README.md
View file @
f4e62d75
...
@@ -28,7 +28,7 @@ MobileNetV2的网络结构主要由两部分组成:特征提取层和分类器
...
@@ -28,7 +28,7 @@ MobileNetV2的网络结构主要由两部分组成:特征提取层和分类器
git
clone
--
recursive
http
:
//
developer
.
hpccube
.
com
/
codes
/
modelzoo
/
mobilenetv2_mmcv
.
git
git
clone
--
recursive
http
:
//
developer
.
hpccube
.
com
/
codes
/
modelzoo
/
mobilenetv2_mmcv
.
git
docker
pull
image
.
sourcefind
.
cn
:
5000
/
dcu
/
admin
/
base
/
pytorch
:
1.10
.
0
-
centos7
.
6
-
dtk
-
22.10
.
1
-
py37
-
latest
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替换
# <your IMAGE ID>用以上拉取的docker的镜像ID替换
docker
run
--
shm
-
size
10
g
--
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
/
mobilenetv2_mmcv
:
/
home
/
mobilenetv2_mmcv
-
it
<
your
IMAGE
ID
>
bash
docker
run
--
shm
-
size
10
g
--
network
=
host
--
name
=
mobilenetv2
--
privileged
--
device
=/
dev
/
kfd
--
device
=/
dev
/
dri
--
group
-
add
video
--
cap
-
add
=
SYS_PTRACE
--
security
-
opt
seccomp
=
unconfined
-
v
$
PWD
/
mobilenetv2_mmcv
:
/
home
/
mobilenetv2_mmcv
-
it
<
your
IMAGE
ID
>
bash
cd
mobilenetv2_mmcv
/
mmclassification
-
mmcv
cd
mobilenetv2_mmcv
/
mmclassification
-
mmcv
pip
install
-
r
requirements
.
txt
pip
install
-
r
requirements
.
txt
...
@@ -39,7 +39,7 @@ pip install -r requirements.txt
...
@@ -39,7 +39,7 @@ pip install -r requirements.txt
```
plaintext
```
plaintext
cd mobilenetv2_mmcv/docker
cd mobilenetv2_mmcv/docker
docker build --no-cache -t mobilenetv2_mmcv:latest .
docker build --no-cache -t mobilenetv2_mmcv:latest .
docker run --rm --shm-size 10g --network=host --name=m
egatron
--privileged --device=/dev/kfd --device=/dev/dri --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -v $PWD/../../mobilenetv2_mmcv:/home/mobilenetv2_mmcv -it
megatron
bash
docker run --rm --shm-size 10g --network=host --name=m
obilenetv2
--privileged --device=/dev/kfd --device=/dev/dri --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -v $PWD/../../mobilenetv2_mmcv:/home/mobilenetv2_mmcv -it
<your IMAGE ID>
bash
# 若遇到Dockerfile启动的方式安装环境需要长时间等待,可注释掉里面的pip安装,启动容器后再安装python库:pip install -r requirements.txt
# 若遇到Dockerfile启动的方式安装环境需要长时间等待,可注释掉里面的pip安装,启动容器后再安装python库:pip install -r requirements.txt
```
```
...
@@ -71,9 +71,15 @@ pip install -r requirements.txt
...
@@ -71,9 +71,15 @@ pip install -r requirements.txt
下载val数据:链接:https://pan.baidu.com/s/1oXsmsYahGVG3uOZ8e535LA?pwd=c3bc 提取码:c3bc 替换ImageNet数据集中的val目录,处理后的数据结构如下:
下载val数据:链接:https://pan.baidu.com/s/1oXsmsYahGVG3uOZ8e535LA?pwd=c3bc 提取码:c3bc 替换ImageNet数据集中的val目录,处理后的数据结构如下:
```
```
├── meta
data
├── train
├──imagenet
├── val
├── meta
├──val.txt
├──train.txt
...
├── train
├── val
```
```
## 训练
## 训练
...
@@ -84,7 +90,11 @@ pip install -r requirements.txt
...
@@ -84,7 +90,11 @@ pip install -r requirements.txt
./mobilenetv2.sh
./mobilenetv2.sh
## 精度
## result

### 精度
测试数据使用的是ImageNet数据集,使用的加速卡是DCU Z100L。
测试数据使用的是ImageNet数据集,使用的加速卡是DCU Z100L。
...
@@ -92,10 +102,6 @@ pip install -r requirements.txt
...
@@ -92,10 +102,6 @@ pip install -r requirements.txt
| :--: | :-----------------------: |
| :--: | :-----------------------: |
| 8 | top1:0.71764;top5:0.90386 |
| 8 | top1:0.71764;top5:0.90386 |
## result

## 应用场景
## 应用场景
### 算法类别
### 算法类别
...
@@ -106,10 +112,10 @@ pip install -r requirements.txt
...
@@ -106,10 +112,10 @@ pip install -r requirements.txt
制造,能源,交通,网安
制造,能源,交通,网安
##
#
源码仓库及问题反馈
## 源码仓库及问题反馈
https://developer.hpccube.com/codes/modelzoo/mobilenetv2_mmcv
https://developer.hpccube.com/codes/modelzoo/mobilenetv2_mmcv
##
#
参考
## 参考
资料
https://github.com/open-mmlab/mmpretrain
https://github.com/open-mmlab/mmpretrain
\ No newline at end of file
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment