I have been using a mobilenet-ssd successfully.
I use the below function to restructure the array. 
    public float[][] floatArray1dto2d(final float[] array, final int rows, final int cols) {
        if (array.length != (rows * cols))
            throw new IllegalArgumentException("Invalid array length");

        float[][] outArray = new float[rows][cols];
        for (int i = 0; i < rows; i++)
            System.arraycopy(array, (i * cols), outArray[i], 0, cols);

        return outArray;
    }