strand2013 / NNIE-lite

⚡️ Using NNIE as simple as using ncnn ⚡️
MIT License
183 stars 50 forks source link

ParseYolov3Feature计算框和类别的得分问题 #13

Open lxgyChen opened 3 years ago

lxgyChen commented 3 years ago

你好!想问下utils.h里ParseYolov3Feature函数里面: 108行float tc = feature.data[cx + (cy feature.width) + (channel + 4) feature.height * feature.width];应该是目标框的得分,然后 123行用Softmax计算每个类别的得分,但是yolov3之后不是用Sigmoid计算每个框的得分吗,这样可以实现多标签分类,同一个框可以说不同类别的物体。我看海思给的yolov3 demo里,在sample_svp_nnie_software.c,SVP_NNIE_Yolov3_GetResult函数里也是用Sigmoid计算框和各类别的得分的