資源簡介
tensorflow利用預訓練的模型提取圖片的特征,以VGG為例,使用slim庫
代碼片段和文件信息
#!/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
???????????????????
評論
共有 條評論