如何在张量流中对不同的图像类型使用不同的损失？

Question

我有两种图像，它们的名称分别以两个不同的字符串开头（例如'a _....'和'b _...'）。我想基于当前正在训练的图像使用不同的损失函数（批量大小为1）。如何包含if条件以对图像使用特定损失？

谢谢！

Answer 1

尝试在python中读取和使用ternary_operators（存在于任何编程语言中）http://book.pythontips.com/en/latest/ternary_operators.html 如果我理解你的话

def los_func_1():
    print 'los_func 1 '

def los_func_2():
    print 'los_func 2'


if __name__ == '__main__':
    use_1 = True
    los_func_1() if use_1 else los_func_2()

此代码将打印

  

los_func 1