从Fast-RCNN获取特征向量

Question

我正在尝试识别和分组相似的图像，我遵循了本教程：https://douglasduhaime.com/posts/identifying-similar-images-with-tensorflow.html。

问题是我使用的是faster_rcnn_resnet_101，发现在SecondStageBoxPredictor之后，fast-rcnn中的特征向量被删除了。我使用https://gist.github.com/markdtw/02ece6b90e75832bd44787c03a664e8d来获取矢量，然后将其删除。

feat_avg = graph.get_tensor_by_name('SecondStageBoxPredictor/AvgPool:0')

np.savetxt('output1/' + "test" + '.npz',feat_vector,delimiter=',')

但是，当我尝试保存矢量时，出现错误：

ValueError: Expected 1D or 2D array, got 4D array instead

我打印了提取的特征向量以查看结果：

    Tensor("SecondStageFeatureExtractor/resnet_v1_101/block4/unit_3/bottleneck_v1/Relu:0", shape=(?, 7, 7, 2048), dtype=float32) 
 [[[[0.         0.         0.         ... 0.         0.
    0.        ]
   [0.         0.         0.         ... 0.         0.
    0.        ]
   [0.         0.         0.         ... 0.         0.
    0.        ]
   ...
   [2.9170244  0.         0.33220196 ... 0.         0.
    0.        ]
   [0.         0.         0.         ... 0.         0.
    0.        ]
   [0.         0.         0.         ... 0.         0.
    0.        ]].....

我是TensorFlow和CV的新手，我想做的是提取特征向量，然后使用TSNE聚类。 我提取的特征向量到底有什么问题

Answer 1

特征向量是4D传感器：[批处理，高度，宽度，通道]。 np.savetxt等待一维或二维数组。您可以将特征向量切片为2D数组，或使用其他功能将其保存为4D。