你好,我这里遇到如下的报错:
Update stage ---- output cls_score and bbox_pred extracting cost time : 6 ms
Update stage ---- postprocess cost time : 0 ms
pscore_window max score is: nan
通过单步调试发现, cls_score_data, bbox_pred_data1,bbox_pred_data2, bbox_pred_data3, bbox_pred_data4都是 NaN
` float* cls_score_data = (float*)cls_score.data;
cls_score_sigmoid.clear();
int cols = cls_score.w;
int rows = cls_score.h;
for (int i = 0; i < cols*rows; i++) // 18 * 18
{
cls_score_sigmoid.push_back(sigmoid(cls_score_data[i]));
}
std::vector<float> pred_x1(cols*rows, 0), pred_y1(cols*rows, 0), pred_x2(cols*rows, 0), pred_y2(cols*rows, 0);
float* bbox_pred_data1 = bbox_pred.channel(0);
float* bbox_pred_data2 = bbox_pred.channel(1);
float* bbox_pred_data3 = bbox_pred.channel(2);
float* bbox_pred_data4 = bbox_pred.channel(3);`
难道是在网络的输出地方就出错了?
ex_neck_head.extract("output.1", cls_score); // [c, w, h] = [1, 18, 18] ex_neck_head.extract("output.2", bbox_pred); // [c, w, h] = [4, 18, 18]