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
Mask rcnn调参
最新推荐文章于 2024-04-22 17:10:01 发布