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 ... ")
Utilities.py 文件源码
python
阅读 30
收藏 0
点赞 0
评论 0
评论列表
文章目录