Отображение изображения в списке словаря с другими элементами - PullRequest
Новая
       17

Отображение изображения в списке словаря с другими элементами

0 голосов
/ 13 июня 2019

Я реализую предварительно обученное обнаружение и сегментацию, предоставленную с сайта pytorch с моим собственным изображением. Когда я печатаю «output», он выдает список словаря, в котором есть информация о полях, масках, метках и т. Д. Я хочу отобразить всю эту информацию на одном изображении, но знаю только, что отображать маску.

Пожалуйста, обратитесь к этому сайту здесь . 

import torchvision
import torchvision.transforms as transforms
import tensorflow as tf
import matplotlib.pyplot as plt
import numpy as np
import PIL
from PIL import Image

model= torchvision.models.detection.maskrcnn_resnet50_fpn(pretrained=True)
model.eval()

image = PIL.Image.open('ClothingAttributeDataset/images/000020.jpg')
image_tensor = torchvision.transforms.functional.to_tensor(image)

output = model([image_tensor])
print(output) # it will give a output of list of dictionary

for i in range(14):
    img = output[0]["masks"]
    img = img.data.cpu().numpy()[i]
    img = np.transpose(img, (1, 2, 0)).reshape(750, 500)
    plt.imsave('objects/img%d' %i, img)
...