Skip to content

icodingc/ImageRetrieval-tf

Repository files navigation

基于TensorFlow以及TF-serving图像检索

1. 抽取特征extract_feature <br>

  • 这里没有训练网络,只是根据ImageNet上预训练Inception_v3模型抽取mixed_8x8x2048b层特征,然后sum_pooling到2048-d。
  • 如想得到更好的特征表示,可以参考DeepRank等文章根据数据集fine-tune训练网络。

2. 运行服务Serving <br>

  • 因为Tensorflow Serving 更新版本导致本项目不能使用。
  • 不过最新版本的Serving更容易学习使用,如感兴趣可以看下我写的其他例子clientexporter以及server

4. 本项目采用的是tensorflow 0.10.0rc0,而且可能不会再更新

下边是几个检索例子