ymir-detectron2 镜像说明文档#

代码仓库#

参考facebook/detectron2 - modelai/ymir-detectron2

镜像地址#

youdaoyzbx/ymir-exectutor:ymir2.0.0-detectron2-cu111-tmi

性能表现#

数据参考detectron2/Model Zoo

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}
}