coco一共有18个关键点代码如下

 

 这是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]]