24、TensorFlow教程--- 使用 TensorFlow 进行图像识别

TensorFlow包含一项特殊的图像识别功能，这些图像存储在特定的文件夹中。使用相对相似的图像，可以轻松实现此逻辑以用于安全目的。

图像识别代码实现的文件夹结构如下所示 −

dataset_image文件夹包含相关的图像，需要加载这些图像。我们将专注于具有我们公司标志的图像识别。图像是使用“load_data.py”脚本加载的，该脚本有助于跟踪其中各种图像识别模块的信息。

import pickle
from sklearn.model_selection import train_test_split
from scipy import misc

import numpy as np
import os

label = os.listdir("dataset_image")
label = label[1:]
dataset = []

for image_label in label:
   images = os.listdir("dataset_image/"+image_label)
   
   for image in images:
      img = misc.imread("dataset_image/"+image_label+"/"+image)
      img = misc.imresize(img, (64, 64))
      dataset.append((img,image_label))
X = []
Y = []

for input,image_label in dataset:
   X.append(input)
   Y.append(labe