tensorflow - 是否可以使用分类器模型的输出分支到不同的模型？

Question

例如，模型 A 有一个输入图像并对它进行分类。假设它的输出维度为 3，例如(0,1,0)。然后根据输出，将图像传递到 Model X、Y 或 Z 中。本质上我是在寻找一个层来执行此条件？

Answer 1

使用 Tensorflow，tf.case() 可能就是您想要的，see doc了解更多信息。

这里是一个例子，假设默认使用模型Z，并且假设条件是互斥的:

condition_for_X = conditionX(output_A)  # e.g. tf.less(tf.reduce_sum(output_A), -1.) 
condition_for_Y = conditionX(output_A)  # e.g. tf.greater_equal(tf.reduce_sum(output_A), 1.) 

def use_X(): return model_X(output_A)
def use_Y(): return model_Y(output_A)
def use_Z(): return model_Z(output_A)
result = tf.case({condition_for_X: use_X, condition_for_Y : use_Y},
                  default=use_Z, exclusive=True)