MASK_RCNN解读3 原始训练模型加载和构建
MASK_RCNN解读3-原始训练FPN模型加载 现在开始进行网络的训练,脚本为: python tools/train_net.py –config-file configs/my_e2e_mask_rcnn_R_50_FPN_1x.yaml MODEL.ROI_BOX_HEAD.NUM_CLASSES 6 SOLVER.IMS_PER_BATCH 4 SOLVER.BASE_LR 0.001 SOLVER.MAX_ITER 36000 SOLVER.STEPS ” (24000, 32000)” TEST.IMS_PER_BATCH 1 MODEL.RPN.FPN_POST_NMS_TOP
用户评论