这是lightweight openpose读代码的出来过程。
最后keypont_maps[-1]是第19个通道是用1-减去18个通道每个像素的最大值。
了解max函数。
import numpy as np a = np.arange(4).reshape((1,4)) print(a) print(a.max(axis=0)) a = np.arange(4).reshape((2,2)) print(a) print(a.max(axis=0)) a = np.arange(8).reshape((2,2,2)) a[0,0,0]=8 print(a) print(a.max(axis=0))[[0 1 2 3]]
[0 1 2 3]
[[0 1]
[2 3]]
[2 3]
[[[8 1]
[2 3]]
[[4 5]
[6 7]]]
[[8 5]
[6 7]]