ymir-detectron2 镜像说明文档#
代码仓库#
镜像地址#
youdaoyzbx/ymir-exectutor:ymir2.0.0-detectron2-cu111-tmi
性能表现#
| Name | lr sched |
train time (s/iter) |
inference time (s/im) |
train mem (GB) |
box AP |
model id | download |
|---|---|---|---|---|---|---|---|
| R50 | 1x | 0.205 | 0.041 | 4.1 | 37.4 | 190397773 | model | metrics |
| R50 | 3x | 0.205 | 0.041 | 4.1 | 38.7 | 190397829 | model | metrics |
| R101 | 3x | 0.291 | 0.054 | 5.2 | 40.4 | 190397697 | model | metrics |
训练参数#
| 超参数 | 默认值 | 类型 | 说明 | 建议 |
|---|---|---|---|---|
| hyper-parameter | default value | type | note | advice |
| batch_size | 2 | 整数 | batch size 大小 | - |
| config_file | configs/COCO-Detection/retinanet_R_50_FPN_1x.yaml | 文件路径 | 配置文件路径 | 参考 configs/COCO-Detection |
| max_iter | 90000 | 整数 | 最大训练次数 | - |
| learning_rate | 0.001 | 浮点数 | 学习率 | - |
| args_options | '' | 字符串 | 命令行参数 | 参考 default_argument_parser |
推理参数#
| 超参数 | 默认值 | 类型 | 说明 | 建议 |
|---|---|---|---|---|
| hyper-parameter | default value | type | note | advice |
| conf_threshold | 0.2 | 浮点数 | 置信度阈值 | 采用默认值 |
挖掘参数#
| 超参数 | 默认值 | 类型 | 说明 | 建议 |
|---|---|---|---|---|
| hyper-parameter | default value | type | note | advice |
| conf_threshold | 0.2 | 浮点数 | 置信度阈值 | 采用默认值 |
引用#
@misc{wu2019detectron2,
author = {Yuxin Wu and Alexander Kirillov and Francisco Massa and
Wan-Yen Lo and Ross Girshick},
title = {Detectron2},
howpublished = {\url{https://github.com/facebookresearch/detectron2}},
year = {2019}
}