strand2013 / NNIE-lite

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

clsInfo应该把返回类别都存入吧 #5

Open northeastsquare opened 3 years ago

northeastsquare commented 3 years ago

https://github.com/RaySue/NNIE-lite/blob/master/examples/GeneralCls.cpp#L111 您好,请问这句push_back, 是否应该放在if前面,存错所有类别得分,index里面保存最大值得索引。

strand2013 commented 3 years ago

https://github.com/RaySue/NNIE-lite/blob/master/examples/GeneralCls.cpp#L111 您好,请问这句push_back, 是否应该放在if前面,存错所有类别得分,index里面保存最大值得索引。

是的,这里疏忽了,抱歉