siriusdemon
commented
4 years ago 在创建网络的时候，设置多类别。比如 COCO 中有 80 类，所以'hm':80
net = get_mobile_net(10, {'hm':80, 'wh':2, 'off':2}, head_conv=24)关键点根据你自己的数据集设置，因为一个点要 x, y 两个坐标，所以设置的时候要 x 2。比如人脸五个关键点 lm: 10
net = get_mobile_net(10, {'hm':1, 'wh':2, 'lm':10, 'off':2}, head_conv=24)
可以检测多个类别吗 比如单层车牌/双层车牌，同时有关键点，大佬能给点思路吗