Visualization of the head load distribution

Hello, thank you very much for your work. I am very interested in it. However, I have a question. In your ablation study, there is a work related to the visualization of head load distribution. Which part of the code does this method correspond to?