def grid_sample(pred_score_map, down_rate=20, topk=512): num_row = pred_score_map.shape[0] // down_rate num_col = pred_score_map.shape[1] // down_rate

这段代码的作用是对输入的预测分数图进行下采样，并返回对应的行列坐标。其中，输入的 pred_score_map 是模型对输入图片的预测分数图，down_rate 是下采样率，topk 是返回的坐标点数。首先，根据 down_rate 计算出下采样后的行数和列数，然后将预测分数图按照下采样后的行列数进行切分，对每一个小块求平均分数，并将其作为该小块的输出分数。最后对输出分数进行排序，取前 topk 个分数最高的坐标，并将这些坐标按行、列的顺序拼接成一个列表返回。