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
MLPerf_MaskRCNN_pytorch
Commits
ca62106f
"src/diffusers/schedulers/scheduling_lcm.py" did not exist on "4c54519e1a640f393ff790a72be38284d4253b45"
Commit
ca62106f
authored
Apr 26, 2023
by
liangjing
Browse files
add model.properties
parent
b30eb11b
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
17 additions
and
3 deletions
+17
-3
README.md
README.md
+8
-3
model.properties
model.properties
+9
-0
No files found.
README.md
View file @
ca62106f
...
...
@@ -54,7 +54,12 @@ COCO数据集的广泛使用使得其成为计算机视觉领域中的一个标
python3 pickle_coco_annotations.py --root <path/to/detectron2/dataset/dir> --ann_file <path/to/coco/annotation/file> --pickle_output_file <path/to/pickled/output/file>
## 训练
### 测试规模
单机8卡进行性能&&精度测试
### 环境配置
提供
[
光源
](
https://www.sourcefind.cn/#/service-details
)
拉取的训练的docker镜像:
*
训练镜像:
...
...
@@ -65,13 +70,13 @@ python依赖安装:
cd ./cocoapi-0.7.0/PythonAPI; python3 setup.py install
### 训练
训练命令
(此处以单机8卡规模为例说明)
:
训练命令:
bash sbatch.sh #输出结果见log.txt
注:可通过修改./maskrcnn_benchmark.bak/config/paths_catalog_dbcluster.py文件按需更改数据读取位置
##
性能和准确率数据
测试
采用上述输入数据,加速卡采用Z100L,可最终达到收敛
精度
要求
##
测试结果
采用上述输入数据,加速卡采用Z100L
*
8
,可最终达到
官方
收敛要求
;
## 历史版本
*
https://developer.hpccube.com/codes/modelzoo/mlperf_maskrcnn
...
...
model.properties
0 → 100644
View file @
ca62106f
# 模型名称
modelName
=
MaskRCNN
# 模型描述
modelDescription
=
MaskRCNN是一种基于Faster R-CNN的物体检测和实例分割模型
# 应用场景(多个标签以英文逗号分割)
appScenario
=
CV
# 框架类型(多个标签以英文逗号分割)
frameType
=
Pytorch
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