472788a1ebc93aa18c844a3da1efb36cbda30bfa,lanenet_model/lanenet_merge_model.py,,,#,157

Before Change

    instance_label = tf.placeholder(dtype=tf.float32, shape=[1, 256, 512, 1], name="label")
    ret = model.compute_loss(input_tensor=input_tensor, binary_label=binary_label,
                             instance_label=instance_label, name="loss")
    print(ret["total_loss"])

After Change

    instance_label = tf.placeholder(dtype=tf.float32, shape=[1, 256, 512, 1], name="label")
    ret = model.compute_loss(input_tensor=input_tensor, binary_label=binary_label,
                             instance_label=instance_label, name="loss")
    for vv in tf.trainable_variables():
        if "bn" in vv.name:
            continue
        print(vv.name)

Instances