tensorflow提取图片特征-

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
“““
Created on Tue Nov  7 10:17:31 2017

@author: hwj
“““

import sys
import os
os.environ[“CUDA_VISIBLE_DEVICES“] = ‘0‘
sys.path.append（‘/home/hwj/tf_practice/models/research/slim‘）
import tensorflow as tf
import tensorflow.contrib.slim as slim

import numpy as np
import nets.vgg as vgg
from datasets import imagenet
from preprocessing import vgg_preprocessing
import scipy.io as scio

‘‘‘
extract specific layer feature of vgg_16 heckpoints_dir and layer name are needed 

‘‘‘


def extract_feature_vgg_16（checkpoints_dir input_image layer）:
    image_size = vgg.vgg_16.default_image_size
    processed_image = vgg_preprocessing.preprocess_image（input_image
                                                         image_size