Mask RCNN save the best segm model

[hyz-xmaster]

Mask RCNN save the best segm model

[hyz-xmaster]

Hi,

To enable saving the best_segm_mAP.pth, basically, I think you need to make following changes:

swa_load_from = 'best_bbox_mAP.pth' -> swa_load_from = 'best_segm_mAP.pth'

if cfg.swa_load_from == 'best_bbox_mAP.pth': ->
if cfg.swa_load_from == 'best_bbox_mAP.pth' or cfg.swa_load_from == 'best_segm_mAP.pth':

eval_hook(val_dataloader, save_best='bbox_mAP', **eval_cfg)) -> eval_hook(val_dataloader, save_best='segm_mAP', **eval_cfg))

val_dataloader, save_best='bbox_mAP', **eval_cfg) -> val_dataloader, save_best='segm_mAP', **eval_cfg)

[yc0619]

Yess, it works. Thanks again!

[yc0619]

Sorry but i would like to ask again, if swa support the cascade mask rcnn, ms rcnn or htc?

[hyz-xmaster]

Hi, although I did not test SWA with those detectors you mentioned, I think SWA training should work with them. Please see the instructions about how to apply SWA training to them.

[yc0619]

cool thanks a lot!!