如何为特定的卷重新配置YOLO2输出层？

Question

您好，dl4j堆栈器有可能获得除5默认参数[x, y, w, h, p]以外的yolo变体，这是我从github上默认 dl4j-examples 回购中获得的东西13 x 13网格锁定图像

...
graphBuilder.addLayer("convolution2d_23",
                new ConvolutionLayer.Builder(1,1)
                        .nIn(1024)
                        .nOut(nBoxes* (5 +nClasses))//don't want the 5 default params always
                        .weightInit(WeightInit.XAVIER)
                        .stride(1,1)
                        .convolutionMode(ConvolutionMode.Same)
                        .weightInit(WeightInit.RELU)
                        .activation(Activation.IDENTITY)
                        .cudnnAlgoMode(cudnnAlgoMode)
                        .build(),
                "activation_22")
                .addLayer("outputs",
                        new Yolo2OutputLayer.Builder()
                                .boundingBoxPriors(priors)
                                .build(),
                        "convolution2d_23")
                .setOutputs("outputs");
graphBuilder.build();
...

我需要重新配置Yolo2OutputLayer或自定义Yolo2OutputLayer类 输出层，能够输出任何特定的音量。目前 我需要输出一个卷13 x 13 x 80，其中一个单元片1 x 1x 80 = 1 x 1 x 2[x, y, w, h, p, c, a0, a1,..., a31]

  

    

其中2等于每个单元的边界框数量

        x-边界框x坐标- 1
        y-边界框y坐标- 1
        w-边界框宽度- 1
        h-边框高度- 1
        p-边界框对象置信度- 1

        c-边界框类（3个类）- 3
        a0...a31-我的自定义参数- 32