Utilities.py 文件源码

def UserInteract(Model):
    global sess, ColorizationModel
    sess = tf.InteractiveSession() 
    ColorizationModel = Model

    print("")
    print("")

    print("         ---- DEEP LEARNING COLORIZATION FOR VISUAL MEDIA ----           ")

    print("")
    print(" Enter (T) if you want to continue Training, or")
    print(" Enter (t) if you want to Test an Image, or")
    userInput = input(" Enter(X) to exit   :   ")
    if(userInput =='T'):
        UserInteract_train()
    elif(userInput =='t'):
        UserInteract_test()
    else:
        print("Exit ... ")