mask-rcnn_r50_fpn_15e_deepfashion.py 663 B

1234567891011121314151617181920212223
  1. _base_ = [
  2. '../_base_/models/mask-rcnn_r50_fpn.py',
  3. '../_base_/datasets/deepfashion.py', '../_base_/schedules/schedule_1x.py',
  4. '../_base_/default_runtime.py'
  5. ]
  6. model = dict(
  7. roi_head=dict(
  8. bbox_head=dict(num_classes=15), mask_head=dict(num_classes=15)))
  9. # runtime settings
  10. max_epochs = 15
  11. train_cfg = dict(
  12. type='EpochBasedTrainLoop', max_epochs=max_epochs, val_interval=1)
  13. param_scheduler = [
  14. dict(
  15. type='LinearLR', start_factor=0.001, by_epoch=False, begin=0, end=500),
  16. dict(
  17. type='MultiStepLR',
  18. begin=0,
  19. end=max_epochs,
  20. by_epoch=True,
  21. milestones=[8, 11],
  22. gamma=0.1)
  23. ]