java - Java 上的 tensorflow : how to perform RGB to BGR operation?

Question

我需要将包含 RGB 图像的 3-D 张量转换为 BGR。

我在网上找到的所有来源都使用 python，它们引用了 java 上不存在或不同的操作:

1. reverse 不接受索引，而是接受 boolean 值作为第二个输入
2. 我找到了 stack/unstack 方法，但 unstack 不存在

Answer 1

Stack 和 Unstack 操作的实际调用方式与之前在其他平台上的调用方式相同，即 Pack 和 Unpack 。 *

因此，要执行将表示 RGB 图像的 Tensor 转换为 BGR 图像的请求操作，可以遵循 unstack/stack 方式。

    public <T> Output<T> rgbToBgr(Output<T> input) {
        return stackReverted(unstack(input));
    }

    // where g is Graph
    public <T> Output<T>[] unstack(Output<T> input) {
        return (Output<T>[]) g.opBuilder("Unpack", "Unpack")
                .addInput(input)
                .setAttr("axis", -1)
                .setAttr("num", 3)
                .build().outputList(0, 3);
    }


    public <T> Output<T> stackReverted(Output<T>[] input){
        GraphOperationBuilder graphOperationBuilder = g.opBuilder("Pack", "Pack")
                .setAttr("axis", -1);
        Output<T>[] newInput = Arrays.copyOf(input, input.length);
        for (int i = 0; i < input.length; i++)
            newInput[i] = input[input.length - 1 - i];
        graphOperationBuilder.addInputList(newInput);
        return graphOperationBuilder.build().output(0);
    }

• 到处都没有踪迹。如果有人能找到 Java 文档，请发布。