20、图像排序与TensorFlow安装指南

图像排序与TensorFlow安装指南

1. 可视化计算图

在浏览器中打开TensorBoard，导航到Graphs选项卡，即可查看计算图。从图中可以快速了解网络中涉及的层类型，例如VGG16神经网络的最后三层是全连接的密集层，分别标记为fc1、fc2和fc3。

2. 图像排序

为了对图像进行排序，我们将使用VGG16代码获取图像的向量表示，从而在排序神经网络中高效地对两张图像进行排序。以衬衫折叠视频为例，我们逐帧处理视频，对图像状态进行排序，这样算法就能判断是否达到了布料折叠的目标。

具体步骤如下：
1. 下载数据集 ：从http://mng.bz/eZsc下载布料折叠数据集，并解压。记住解压位置，在代码中用 DATASET_DIR 表示。
2. 导入相关库 ：在新的源文件中导入所需的Python库。

import tensorflow as tf
import numpy as np
from vgg16 import vgg16
import glob, os
from scipy.misc import imread, imresize

3. 准备训练数据 ：对于每个视频，我们记录其第一帧和最后一帧图像，假设最后一帧图像比第一帧图像更受偏好，以此来训练排序算法。