NikolasEnt / soccernet-calibration-sportlight

SoccerNet@CVPR | 1st place solution for Camera Calibration Challenge 2023
31 stars 3 forks source link

How to determine sigma for heatmap? #3

Closed Wang-Yu-Qing closed 9 months ago

Wang-Yu-Qing commented 9 months ago

I found the sigma is fixed as value 3 for image of size [270, 480] in train_config.yaml:

loss:
  num_refinement_stages: ${model.params.nn_module.num_refinement_stages}
  stride: 2
  sigma: 3.0
  pred_size: [270, 480]

If the image size is [h, w], how to determine a good sigma? I read from the solution post that it should be 2px, what is this value exactly? Thanks!

NikolasEnt commented 9 months ago

Hi, @Wang-Yu-Qing, thank you for your interest in our solution and spotting the issue. Indeed, sigma should be 3 in the config, I've checked the best model config file. The value was optimized during parameter search during the challenge time. I've updated everything accordingly.