Mask rcnn调参

1.学习率 我是设的0.001
2.训练轮数 如果是之前训好的模型,那么20/40,训到loss降到0.6左右应该比较合适
3.MIN_DIM,短边的长度,我这里是用的原图训,就960了,长边1280,要求能整除64
4.TRAIN_ROIS_PER_IMAGES。这个比较关键,训练和测试不一样。这个值是指提取的ROI的数量,也就是候选框的数量,最好先大概看一下一张图上有多少个目标,然后x3。如果训练用了crop,则训练和测试这个参数要做对应调整
5.IMAGE_RESIZE_MODE,一般用默认的“square”,如果要做random crop就改为"crop",会从原图上crop下MIN_DIMxMIN_DIM大小的图训练,Mask也会做相应crop

  • 1
    点赞
  • 18
    收藏
    觉得还不错? 一键收藏
  • 3
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值