Tensorflow：21个项目（导出模型并预测单张图片）

最新推荐文章于 2022-12-03 11:16:29 发布

马小歪丷

最新推荐文章于 2022-12-03 11:16:29 发布

阅读量1k

点赞数

文章标签： tensorflow

本文链接：https://blog.csdn.net/weixin_41829234/article/details/89914591

版权

本文介绍了如何使用TensorFlow进行项目实践，包括模型的导出和对单张图片的预测过程，详细解析了从训练到应用的关键步骤。

摘要由CSDN通过智能技术生成

参考：tensorflow：21个项目（何之源）

import numpy as np
import os
import six.moves.urllib as urllib
import sys
import tarfile
import tensorflow as tf
import zipfile

from collections import defaultdict
from io import StringIO
from matplotlib import pyplot as plt
from PIL import Image

# This is needed to display the images.
%matplotlib inline

# This is needed since the notebook is stored in the object_detection folder.
sys.path.append("..")

from utils import label_map_util

from utils import visualization_utils as vis_util

PATH_TO_CKPT = 'voc/export/frozen_inference_graph.pb'

# List of the strings that is used to add correct label for each box.
PATH_TO_LABELS = "voc/pascal_label_map.pbtxt"

NUM_CLASSES = 20

detection_graph = tf.Graph()
with detection_graph.as_default():
  od_graph_def = tf.GraphDef()
  with tf.gfile.GFile(PATH_TO_CKPT, 'rb') as fid:
    seria