ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

MSSD模型训练config文件参数配置解读

2021-01-27 11:35:58  阅读:284  来源: 互联网

标签:optimizer 训练 decay max MSSD 解读 参数 config


参考链接:

object detection api调参详解(兼SSD算法参数详解)

https://copyfuture.com/blogs-details/20200630233052263i2qwtu5rloywtv1

文章详细介绍了ssd_mobilenet_v1_pets.config中的参数含义与配置建议,大部分还是正确的可供参考。

这与我训练中使用的ssd_mobilenet_v2_coco.config文件基本类似。

下面简单补充几点:

1,anchor_generator参数的配置。

 其中默认参数如下:  min_scale: 0.2    max_scale: 0.95 修改建议:对训练集中的box宽度进行统计,进行相应修改。   2, post_processing  该部分参数也可以根据需要调整。 batch_non_max_suppression {         score_threshold: 1e-8         iou_threshold: 0.6         max_detections_per_class: 100 #每个类别可保留的检测框的最大数目。         max_total_detections: 100 #所有类别可保留的检测框的最大数目。       }       score_converter: SIGMOID   # 检测分数的转换器类型选择   3,train_config: 最重要的修改部分:训练参数 {   batch_size: 48 # 越大越好,只要你的算力和显存大小满足。   optimizer {     rms_prop_optimizer: {  #优化器的选择,还可以尝试使用,       learning_rate: {         exponential_decay_learning_rate { #学习率的变化,除了指数型变化外,还有momentum_optimizer,adam_optimizer这两种优化方式。           initial_learning_rate: 0.004# 初始学习率           decay_steps: 800720 #学习率变化周期           decay_factor: 0.95 #衰减率         }       }       momentum_optimizer_value: 0.9   # 冲量  在学习率比较小时,较大的冲量(冲量不超过1)可以加速收敛。       decay: 0.9       epsilon: 1.0     }   }   4,fine_tune_checkpoint 设置预训练模型的路径,用于精调,并非用于0步训练时使用。
待补充ing  

标签:optimizer,训练,decay,max,MSSD,解读,参数,config
来源: https://www.cnblogs.com/vincent212212/p/14334051.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有