神剑山庄资源网 Design By www.hcban.com
本文实例为大家分享了Tensorflow之MNIST CNN实现并保存、加载模型的具体代码,供大家参考,具体内容如下
废话不说,直接上代码
# TensorFlow and tf.keras import tensorflow as tf from tensorflow import keras # Helper libraries import numpy as np import matplotlib.pyplot as plt import os #download the data mnist = keras.datasets.mnist (train_images, train_labels), (test_images, test_labels) = mnist.load_data() class_names = ['0', '1', '2', '3', '4', '5', '6', '7', '8', '9'] train_images = train_images / 255.0 test_images = test_images / 255.0 def create_model(): # It's necessary to give the input_shape,or it will fail when you load the model # The error will be like : You are trying to load the 4 layer models to the 0 layer model = keras.Sequential([ keras.layers.Conv2D(32,[5,5], activation=tf.nn.relu,input_shape = (28,28,1)), keras.layers.MaxPool2D(), keras.layers.Conv2D(64,[7,7], activation=tf.nn.relu), keras.layers.MaxPool2D(), keras.layers.Flatten(), keras.layers.Dense(576, activation=tf.nn.relu), keras.layers.Dense(10, activation=tf.nn.softmax) ]) model.compile(optimizer=tf.train.AdamOptimizer(), loss='sparse_categorical_crossentropy', metrics=['accuracy']) return model #reshape the shape before using it, for that the input of cnn is 4 dimensions train_images = np.reshape(train_images,[-1,28,28,1]) test_images = np.reshape(test_images,[-1,28,28,1]) #train model = create_model() model.fit(train_images, train_labels, epochs=4) #save the model model.save('my_model.h5') #Evaluate test_loss, test_acc = model.evaluate(test_images, test_labels,verbose = 0) print('Test accuracy:', test_acc)
模型保存后,自己手写了几张图片,放在文件夹C:\pythonp\testdir2下,开始测试
#Load the model new_model = keras.models.load_model('my_model.h5') new_model.compile(optimizer=tf.train.AdamOptimizer(), loss='sparse_categorical_crossentropy', metrics=['accuracy']) new_model.summary() #Evaluate # test_loss, test_acc = new_model.evaluate(test_images, test_labels) # print('Test accuracy:', test_acc) #Predicte mypath = 'C:\\pythonp\\testdir2' def getimg(mypath): listdir = os.listdir(mypath) imgs = [] for p in listdir: img = plt.imread(mypath+'\\'+p) # I save the picture that I draw myself under Windows, but the saved picture's # encode style is just opposite with the experiment data, so I transfer it with # this line. img = np.abs(img/255-1) imgs.append(img[:,:,0]) return np.array(imgs),len(imgs) imgs = getimg(mypath) test_images = np.reshape(imgs[0],[-1,28,28,1]) predictions = new_model.predict(test_images) plt.figure() for i in range(imgs[1]): c = np.argmax(predictions[i]) plt.subplot(3,3,i+1) plt.xticks([]) plt.yticks([]) plt.imshow(test_images[i,:,:,0]) plt.title(class_names[c]) plt.show()
测试结果
自己手写的图片截的时候要注意,空白部分尽量不要太大,否则测试结果就呵呵了
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持。
神剑山庄资源网 Design By www.hcban.com
神剑山庄资源网
免责声明:本站文章均来自网站采集或用户投稿,网站不提供任何软件下载或自行开发的软件!
如有用户或公司发现本站内容信息存在侵权行为,请邮件告知! 858582#qq.com
神剑山庄资源网 Design By www.hcban.com
暂无Tensorflow之MNIST CNN实现并保存、加载模型的评论...
《魔兽世界》大逃杀!60人新游玩模式《强袭风暴》3月21日上线
暴雪近日发布了《魔兽世界》10.2.6 更新内容,新游玩模式《强袭风暴》即将于3月21 日在亚服上线,届时玩家将前往阿拉希高地展开一场 60 人大逃杀对战。
艾泽拉斯的冒险者已经征服了艾泽拉斯的大地及遥远的彼岸。他们在对抗世界上最致命的敌人时展现出过人的手腕,并且成功阻止终结宇宙等级的威胁。当他们在为即将于《魔兽世界》资料片《地心之战》中来袭的萨拉塔斯势力做战斗准备时,他们还需要在熟悉的阿拉希高地面对一个全新的敌人──那就是彼此。在《巨龙崛起》10.2.6 更新的《强袭风暴》中,玩家将会进入一个全新的海盗主题大逃杀式限时活动,其中包含极高的风险和史诗级的奖励。
《强袭风暴》不是普通的战场,作为一个独立于主游戏之外的活动,玩家可以用大逃杀的风格来体验《魔兽世界》,不分职业、不分装备(除了你在赛局中捡到的),光是技巧和战略的强弱之分就能决定出谁才是能坚持到最后的赢家。本次活动将会开放单人和双人模式,玩家在加入海盗主题的预赛大厅区域前,可以从强袭风暴角色画面新增好友。游玩游戏将可以累计名望轨迹,《巨龙崛起》和《魔兽世界:巫妖王之怒 经典版》的玩家都可以获得奖励。
更新日志
2024年09月29日
2024年09月29日
- 孙悦.1996-伙伴【正大国际】【WAV+CUE】
- 纪钧瀚《钢琴阅读时光 雨中书店聆听轻音乐》[FLAC/分轨][399.62MB]
- 证声音乐图书馆《走向自然 疗心爵士乐》[320K/MP3][87.4MB]
- 证声音乐图书馆《走向自然 疗心爵士乐》[FLAC/分轨][184.94MB]
- 陈慧娴.2018-Priscilla-Ism演唱会3CD(2024环球红馆40复刻系列)【环球】【WAV+CUE】
- 郑秀文.1999-我应该得到(国)【华纳】【WAV+CUE】
- 陈家慧.2011-钢琴酒吧2CD【龙吟唱片】【WAV+CUE】
- 证声音乐图书馆《雨季 蓝调吉他 Rainy Blues》[320K/MP3][45.01MB]
- 证声音乐图书馆《雨季 蓝调吉他 Rainy Blues》[FLAC/分轨][109.13MB]
- 赞多《序章》[320K/MP3][45.54MB]
- 许巍.2004-每一刻都是崭新的【步升大风】【WAV+CUE】
- 群星.2024-四方馆影视原声带【韶愔音乐】【FLAC分轨】
- 陈雷.1997-安锁咧【金圆唱片】【WAV+CUE】
- 关淑怡.2013-MY.FAVORITE.SK.3CD【环球】【WAV+CUE】
- Sweety.2006-花言乔语【丰华】【WAV+CUE】