DDTL_resnet.py 文件源码

def resnet_model(image,reuse):
    with tf.variable_scope("model",reuse=reuse):
        with slim.arg_scope(resnet_v1.resnet_arg_scope()):
            outputs,_ = resnet_v1.resnet_v1_50(image)
            #outputs,_ = inception_resnet_v2(image)
            outputs = slim.flatten(outputs)
            outputs = slim.fully_connected(outputs,256)
            logits = slim.fully_connected(outputs,num_classes,activation_fn=None)
    return outputs,logits

# -------- train -----------------------------------