VGG16图像分类源代码、测试图片

VGG16图像分类源代码、测试图片下载。

#coding:utf-8
# 实现图像识别
import numpy as np
import tensorflow as tf
import matplotlib.pyplot as plt
import vgg16
import utils
from Nclasses import labels

img_path = input（‘Input the path and image name:‘）
img_ready = utils.load_image（img_path） #（1 224 224 3）

fig=plt.figure（u“Top-5 预测结果“） 

with tf.Session（） as sess:
    images = tf.placeholder（tf.float32 [1 224 224 3]）
    #通过vgg16的初始化函数 实例化vgg，读出了保存在npy文件中的模型参数
    vgg = vgg16.Vgg16（） 
    vgg.forward（images） #复现神经网络结构
    # 得出1000个分类的概率分布
    probability = sess.run（vgg.prob feed_dict={images:img_ready}）
    #概率最高的5个 概率索引值存入top5
    top5 = np.argsort（probability[0]）[-1:-6:-1]
    print（“top5:“top5）
    values = []
    bar_label = []  #标签字典对应的值 5个物种的名称
    for n i in enumerate（top5）: 
        print（“n:“n）
        print（“i:“i）
        values.append（probability[0][i]） 
        bar_label.append（labels[i]） 
        print（i “:“ labels[i] “----“ utils.percent（probability[0][i]） ）
        
    ax = fig.add_subplot（111） 
    ax.bar（range（len（values）） values tick_label=bar_label width=0.5 fc=‘g‘）
    ax.set_ylabel（u‘probabilityit‘） 
    ax.set_title（u‘Top-5‘） 
    for ab in zip（range（len（values）） values）:
        ax.text（a b+0.0005 utils.percent（b） ha=‘center‘ va = ‘bottom‘ fontsize=7）   
    plt.show（） 


    

 属性            大小     日期    时间   名称
----------- ---------  ---------- -----  ----

     文件       1462  2020-06-07 13:42  app.py

     文件      31525  2018-04-29 11:39  Nclasses.py

     文件       1134  2020-06-07 13:44  utils.py

     文件       3648  2020-06-07 13:47  vgg16.py

     文件     109047  2018-04-29 11:39  pic\0.jpg

     文件       9483  2018-04-29 11:39  pic\1.jpg

     文件      38943  2018-04-29 11:39  pic\2.jpg

     文件      15128  2018-04-29 11:39  pic\3.jpg

     文件      23395  2018-04-29 11:39  pic\4.jpg

     文件     172056  2018-04-29 11:39  pic\5.jpg

     文件     149931  2018-04-29 11:39  pic\6.jpg

     文件     143844  2018-04-29 11:39  pic\7.jpg

     文件      31002  2018-04-29 11:39  pic\8.jpg

     文件      13156  2018-04-29 11:39  pic\9.jpg

     文件      92457  2018-04-29 11:39  pic\a.jpg

     文件      70870  2018-04-29 11:39  pic\b.jpg

     文件     279227  2018-04-29 11:39  pic\c.jpg

     文件      52562  2018-04-29 11:39  pic\d.jpg

     目录          0  2018-04-29 11:39  pic

----------- ---------  ---------- -----  ----

              1238870                    19