基于此示例，我尝试使用tensorflow中的图像识别来从多个图像中获取标记： https://github.com/tensorflow/tensorflow/blob/master/tensorflow/models/image/imagenet/classify_image.py
我遇到的问题是，每次调用模块时，进程都会增加内存并且从不释放，就像内存泄漏一样。
for sPhotoPath in lsPhotoFiles:
print('Process memory: '+str(memory() / 1e6)+' MB')
print('Recognizing image: '+sPhotoPath)
ldTags = imagerecognition.getImageTags(sPhotoPath)
print(ldTags)
该模块类似于：
def getImageTags(sFilePath):
lsTags = run_inference_on_image(sFilePath)
return lsTags
使用代码中唯一修改过的部分：
def run_inference_on_image(image):
"""Runs inference on an image.
Args:
image: Image file name.
Returns:
Nothing
"""
if not tf.gfile.Exists(image):
tf.logging.fatal('File does not exist %s', image)
image_data = tf.gfile.FastGFile(image, 'rb').read()
# Creates graph from saved GraphDef.
create_graph()
with tf.Session() as sess:
# Some useful tensors:
# 'softmax:0': A tensor containing the normalized prediction across
# 1000 labels.
# 'pool_3:0': A tensor containing the next-to-last layer containing 2048
# float description of the image.
# 'DecodeJpeg/contents:0': A tensor containing a string providing JPEG
# encoding of the image.
# Runs the softmax tensor by feeding the image_data as input to the graph.
softmax_tensor = sess.graph.get_tensor_by_name('softmax:0')
predictions = sess.run(softmax_tensor,
{'DecodeJpeg/contents:0': image_data})
predictions = np.squeeze(predictions)
# Creates node ID --> English string lookup.
node_lookup = NodeLookup()
top_k = predictions.argsort()[-FLAGS.num_top_predictions:][::-1]
lsTags = []
for node_id in top_k:
human_string = node_lookup.id_to_string(node_id) # 'dock, dockage, docking facility'
score = predictions[node_id]
# print('%s (score = %.5f)' % (human_string, score))
lsHumanString = map((lambda s: s.strip()), human_string.split(','))
sTag = lsHumanString[0]
lsTags.append(sTag)
return lsTags
输出如下：
nil@Lenovo-Z50-70:~/miscellaneous/duplicate/image_recognition$ python test.py Process memory: 911.888384 MB Recognizing image: /home/nil/miscellaneous/duplicate/image_recognition/../image_matching/photos/group1031/airbnb/2823240_23342502.jpg >> Downloading inception-2015-12-05.tgz 100.0% Succesfully downloaded inception-2015-12-05.tgz 88931400 bytes. W tensorflow/core/framework/op_def_util.cc:332] Op BatchNormWithGlobalNormalization is deprecated. It will cease to work in GraphDef version 9. Use tf.nn.batch_normalization(). ['chiffonier', 'wardrobe', 'file', 'studio couch', 'four-poster'] Process memory: 1750.99904 MB Recognizing image: /home/nil/miscellaneous/duplicate/image_recognition/../image_matching/photos/group1031/airbnb/40118_479346.jpg ['restaurant', 'bookshop', 'tobacco shop', 'library', 'bakery'] Process memory: 1812.754432 MB Recognizing image: /home/nil/miscellaneous/duplicate/image_recognition/../image_matching/photos/group1031/airbnb/2823240_23342503.jpg ['shoji', 'four-poster', 'window shade', 'sliding door', 'studio couch'] Process memory: 1916.932096 MB Recognizing image: /home/nil/miscellaneous/duplicate/image_recognition/../image_matching/photos/group1031/airbnb/40118_479343.jpg ['studio couch', 'four-poster', 'quilt', 'wardrobe', 'mosquito net'] Process memory: 2024.63232 MB Recognizing image: /home/nil/miscellaneous/duplicate/image_recognition/../image_matching/photos/group1031/airbnb/2823240_23342512.jpg ['space bar', 'typewriter keyboard', 'table lamp', 'lampshade', 'printer'] Process memory: 2131.304448 MB Recognizing image: /home/nil/miscellaneous/duplicate/image_recognition/../image_matching/photos/group1031/airbnb/40118_479345.jpg ['shoji', 'four-poster', 'window shade', 'sliding door', 'studio couch'] Process memory: 2240.454656 MB Recognizing image: /home/nil/miscellaneous/duplicate/image_recognition/../image_matching/photos/group1031/airbnb/40118_479349.jpg ['sliding door', 'wardrobe', 'entertainment center', 'medicine chest', 'shoji'] Process memory: 2347.286528 MB Recognizing image: /home/nil/miscellaneous/duplicate/image_recognition/../image_matching/photos/group1031/airbnb/2823240_23342505.jpg ['wardrobe', 'shoji', 'window shade', 'studio couch', 'quilt'] Process memory: 4010.266624 MB Recognizing image: /home/nil/miscellaneous/duplicate/image_recognition/../image_matching/photos/group1031/airbnb/2823240_23342511.jpg ['chiffonier', 'wardrobe', 'file', 'chest', 'crib'] Process memory: 4312.395776 MB Recognizing image: /home/nil/miscellaneous/duplicate/image_recognition/../image_matching/photos/group1031/airbnb/2823240_23342508.jpg ['moving van', 'boathouse', 'sliding door', 'butcher shop', 'forklift'] Process memory: 4613.394432 MB Recognizing image: /home/nil/miscellaneous/duplicate/image_recognition/../image_matching/photos/group1031/airbnb/2823240_23342498.jpg ['dock', 'lakeside', 'seashore', 'pier', 'liner'] Process memory: 4915.449856 MB Recognizing image: /home/nil/miscellaneous/duplicate/image_recognition/../image_matching/photos/group1031/airbnb/2823240_23342509.jpg ['studio couch', 'dining table', 'four-poster', 'patio', 'china cabinet'] Process memory: 5217.521664 MB Recognizing image: /home/nil/miscellaneous/duplicate/image_recognition/../image_matching/photos/group1031/airbnb/40118_479344.jpg ['dock', 'lakeside', 'seashore', 'container ship', 'liner'] Process memory: 5518.82752 MB Recognizing image: /home/nil/miscellaneous/duplicate/image_recognition/../image_matching/photos/group1031/airbnb/40118_479347.jpg ['restaurant', 'tobacco shop', 'microwave', 'shoji', 'home theater'] Process memory: 5820.383232 MB Recognizing image: /home/nil/miscellaneous/duplicate/image_recognition/../image_matching/photos/group1031/airbnb/2823240_23342510.jpg ['studio couch', 'crate', 'wardrobe', 'four-poster', 'chiffonier'] Process memory: 6122.430464 MB Recognizing image: /home/nil/miscellaneous/duplicate/image_recognition/../image_matching/photos/group1031/airbnb/2823240_23342504.jpg ['restaurant', 'altar', 'tobacco shop', 'sewing machine', 'desk'] Process memory: 6423.30624 MB Recognizing image: /home/nil/miscellaneous/duplicate/image_recognition/../image_matching/photos/group1031/airbnb/40118_479350.jpg ['studio couch', 'four-poster', 'dining table', 'wardrobe', 'entertainment center'] Process memory: 6725.758976 MB Recognizing image: /home/nil/miscellaneous/duplicate/image_recognition/../image_matching/photos/group1031/airbnb/2823240_23342501.jpg ['studio couch', 'four-poster', 'quilt', 'wardrobe', 'mosquito net'] Process memory: 7026.778112 MB Recognizing image: /home/nil/miscellaneous/duplicate/image_recognition/../image_matching/photos/group1031/airbnb/40118_479352.jpg ['wardrobe', 'shoji', 'window shade', 'studio couch', 'quilt'] Process memory: 7328.5632 MB ...
我检查了会话和图表都被
with包裹起来
如何在主循环中每次迭代后释放内存？感谢