example-head-loss_hrnet-w32_8xb64-210e_coco-256x192.py 518 B

123456789101112131415
  1. # Directly inherit the entire recipe you want to use.
  2. _base_ = 'mmpose::body_2d_keypoint/topdown_heatmap/coco/' \
  3. 'td-hm_hrnet-w32_8xb64-210e_coco-256x192.py'
  4. # This line is to import your own modules.
  5. custom_imports = dict(imports='models')
  6. # Modify the model to use your own head and loss.
  7. _base_['model']['head'] = dict(
  8. type='ExampleHead',
  9. in_channels=32,
  10. out_channels=17,
  11. deconv_out_channels=None,
  12. loss=dict(type='ExampleLoss', use_target_weight=True),
  13. decoder=_base_['codec'])