(locs/hmaps_rescale, hsz, 1, conf.label_blur_rad)
              return [hmaps]
    </code></pre>

    <h3><a id="pb_convert_preds_to_locs"> Defining convert_preds_to_locs function </a></h3>
        Inputs<br>
        <ul>
            <li> pred: Output of network as python/numpy arrays
        </ul>
        Outputs<br>
        <ul>
            <li>locs: 2D landmark locations as B x N x 2
        </ul>

        Override this function to write your function to convert the networks output (as numpy array) to lo