python之x.astype

import numpy as np

x = np.array([1, True, False])
b = x.astype(int)
print(b)

[1 1 0]

这个函数是在人体姿态估计中获取最大热图值使用到

#######这里的目的是将1*4096的预测热图值的矩阵变成64*64热图尺寸的形式
   preds[:, :, 0] = (preds[:, :, 0]) % width
    preds[:, :, 1] = np.floor((preds[:, :, 1]) / width)
#####################筛检出不可见v=0的关节点
    pred_mask = np.tile(np.greater(maxvals, 0.0), (1, 1, 2))
    pred_mask = pred_mask.astype(np.float32)#########这个代码的作用是将False或者True转变成0或者1