Как получить матрицу пикселей распознанных объектов при классификации изображения с помощью Tensorflow?

Question

Я хочу получить пиксельную матрицу объектов внутри изображения, когда изображение классифицируется Tensorflow (classify_image.py).

Иными словами, распознанные объекты должны быть сначала сегментированы. . на картинке есть компьютер, я хочу получить все пиксели, принадлежащие компьютеру.

Но до сих пор я не могу найти образец из учебника Tensorflow.

Единственное, что я могу получить, это результаты распознавания через образец кода Tensorflow.

например.

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] 
for node_id in top_k: 
    human_string = node_lookup.id_to_string(node_id) 
    score = predictions[node_id] 
    print('%s (score = %.5f)' % (human_string, score)) 

Есть ли у кого-то идеи? Это возможно?

Answer 1

Вот некоторые ссылки на модели для сегментации изображений с использованием TensorFlow:

• TensorBox
• DeConvNet
• Segmentation Toolkit