猫狗大战——基于TensorFlow的猫狗识别（5）

最新推荐文章于 2024-01-29 00:46:24 发布

i小玉

最新推荐文章于 2024-01-29 00:46:24 发布

阅读量1.7k

点赞数 7

分类专栏： project 文章标签：深度学习 tensorflow python 机器学习人工智能

本文链接：https://blog.csdn.net/qq_33828738/article/details/106037510

版权

微信公众号：龙跃十二

我是小玉，一个平平无奇的小天才！

悄悄说，github上出了源码详解还有思维导图，答辩PPT，赶快去star！

我的github：https://github.com/ixiaoyu-tech/xiaoyu-project

在前边，我们将训练好的模型存储在指定路径下，这个时候我们的项目也就要接近尾声了，接下来我们需要的就是导入一副图像，做最后的图像测试。

文件名：test.py

代码详解：

def evaluate_one_image():     
image_array=get_one_image_file("./data/test/68.jpg") #测试集路径，自己更改一下

with tf.Graph().as_default():        
    BATCH_SIZE = 1   # 获取一张图片        
    N_CLASSES = 2  #二分类         
    image = tf.cast(image_array, tf.float32)        
    image = tf