numpy的tips合集

整理使用numpy遇到的问题，后续可能加入其他科学计算库。

bool与np.bool_

在输出json时遇到json不认识numpy的bool_，需要做一个bool的强制转换，此外json中null = None

arr=np.array([1,2])
print(type(arr[1]>arr[0]))
# numpy.bool_

对img mask的操作

mask = []
mask = np.repeat(mask, 3, axis=0).transpose((1, 2, 0)).astype(np.uint8)
mask = mask * 255
mask = mask[np.newaxis, :, :]
mask = mask[None, :, :]