yolox-pose_l_4xb64-300e_coco.py 536 B

123456789101112131415161718
  1. _base_ = ['./yolox-pose_s_8xb32-300e_coco.py']
  2. # model settings
  3. model = dict(
  4. init_cfg=dict(checkpoint='https://download.openmmlab.com/mmyolo/v0/yolox/'
  5. 'yolox_l_fast_8xb8-300e_coco/yolox_l_fast_8xb8-300e_'
  6. 'coco_20230213_160715-c731eb1c.pth'),
  7. backbone=dict(
  8. deepen_factor=1.0,
  9. widen_factor=1.0,
  10. ),
  11. neck=dict(
  12. deepen_factor=1.0,
  13. widen_factor=1.0,
  14. ),
  15. bbox_head=dict(head_module=dict(widen_factor=1.0)))
  16. train_dataloader = dict(batch_size=64)