自己的SSD网络跑出来结果有些问题

生成模型是采用ruyistudio，一开始拿自己的caffemodel和ruyistudio例程里的prototxt，生成.wk时是有些问题的，因为类别参数不对，不是21类，是2类。
所以我后来把训练时的prototxt改成了ssd例程的样子，成功生成了.wk文件。

在代码中，我把软件初始化的参数修改了下，如下：
HI_U32 my_class_num = 2; // 21

    /*Set Softmax Parameters*/
    // 类别相关
    pstSoftWareParam->u32SoftMaxInHeight = my_class_num; // 类数 21
    pstSoftWareParam->au32SoftMaxInChn[0] = 121296; // 84*38*38 类别预测的通道数
    pstSoftWareParam->au32SoftMaxInChn[1] = 45486; // 126*19*19
    pstSoftWareParam->au32SoftMaxInChn[2] = 12600; // 126*10*10
    pstSoftWareParam->au32SoftMaxInChn[3] = 3150; // 126*5*5
    pstSoftWareParam->au32SoftMaxInChn[4] = 756; // 84*3*3
    pstSoftWareParam->au32SoftMaxInChn[5] = 84; // 84*1*1

    //
    pstSoftWareParam->u32ConcatNum = 6; // 类别 6组合并到
    pstSoftWareParam->u32SoftMaxOutWidth = 1;
    pstSoftWareParam->u32SoftMaxOutHeight = my_class_num; // 21;
    pstSoftWareParam->u32SoftMaxOutChn = 8732;

    /*Set DetectionOut Parameters*/
    pstSoftWareParam->u32ClassNum = my_class_num; // 21;
    pstSoftWareParam->u32TopK = 400; // 取前top 400个？
    pstSoftWareParam->u32KeepTopK = 200;
    pstSoftWareParam->u32NmsThresh = (HI_U16)(0.3f*SAMPLE_SVP_NNIE_QUANT_BASE);
    pstSoftWareParam->u32ConfThresh = 1;
    pstSoftWareParam->au32DetectInputChn[0] = 23104; // 位置预测通道数  38*38*4*4
    pstSoftWareParam->au32DetectInputChn[1] = 8664; // 19*19*6*4
    pstSoftWareParam->au32DetectInputChn[2] = 2400;
    pstSoftWareParam->au32DetectInputChn[3] = 600;
    pstSoftWareParam->au32DetectInputChn[4] = 144;
    pstSoftWareParam->au32DetectInputChn[5] = 16;

其中，21全都换成了2.
最后结果是这样的：
[img]https://gitee.com/ig__wxz/image_2019/raw/master/2019_7th_mounth/false-result-ssd.png[/img]

同样的图，我用例程的ssd跑，结果是这样的[img]https://gitee.com/ig__wxz/image_2019/raw/master/2019_7th_mounth/ssd-sample-boat-result.png[/img]

请问有大佬知道这个问题怎么解决吗

本帖最后由 17-8559A 于 2019-7-30 15:01 编辑

醉了，，，问题解决了，刚刚把参数列表又扫了一遍，发现不只要改21这个数，pstSoftWareParam->au32SoftMaxInChn这个数组里的数也都要改成对应的参数

贴一个检测结果,下面是自己训练网络的结果：
[img]https://gitee.com/ig__wxz/image_2019/raw/master/2019_7th_mounth/my-design-result.png[/img]

下面是例程ssd的检测结果：
[img]https://gitee.com/ig__wxz/image_2019/raw/master/2019_7th_mounth/sample-ssd-result.png[/img]

虽然结果差不多，但有一点很奇怪
我是用的里的clock记的时间，例程中的ssd跑15FPS，而我自己ssd跑100FPS？模型还都是一样的，只是类别数量不一样。。。

clock测得是cpu的时间，有问题，于是重新用一种方法测了下时间，这次测出来的应该是真实时间了：
[img]https://gitee.com/ig__wxz/image_2019/raw/master/2019_7th_mounth/sampleSSD-real-time.png[/img]
[img]https://gitee.com/ig__wxz/image_2019/raw/master/2019_7th_mounth/MyDesign-real-time.png[/img]

感谢大神分享，天睿视迅也在研发这个

谢谢分享，正遇到此问题