yolox-pose_m_4xb64-300e_coco.py 565 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_m_fast_8xb32-300e-rtmdet-hyp_coco/yolox_m_fast_8xb32'
  6. '-300e-rtmdet-hyp_coco_20230210_144328-e657e182.pth'),
  7. backbone=dict(
  8. deepen_factor=0.67,
  9. widen_factor=0.75,
  10. ),
  11. neck=dict(
  12. deepen_factor=0.67,
  13. widen_factor=0.75,
  14. ),
  15. bbox_head=dict(head_module=dict(widen_factor=0.75)))
  16. train_dataloader = dict(batch_size=64)