Tensorflow实现图像分类

Tensorflow实现图像分类 利用tensorflow做图像分类的训练,主要分为读取图像,训练批次,定义模型,训练,保存模型,模型调用几步。 首先是读取图像,利用python中os模块的listdir方法来读取图像。在这里可以把每一类的图像放到同一文件夹中,文件夹的名字可以直接作为图像的标签,最好是用数字代替,图像的名字最好也是数字,这样写代码比较方便,如图: 读取图像的代码如下: 接下来要做
相关文章
相关标签/搜索