!pip install --upgrade openpifpaf==0.10.1 import io import numpy as np import openpifpaf import PIL import requests import torch print(openpifpaf.__version__) print(torch.__version__) image_response = requests.get('https://i.pinimg.com/originals/8e/30/a6/8e30a6c50bcf6c3dd8fc0ed025df48f4.png') pil_im = PIL.Image.open(io.BytesIO(image_response.content)).convert('RGB') im = np.asarray(pil_im) with openpifpaf.show.image_canvas(im) as ax: pass net_cpu, _ = openpifpaf.network.factory(checkpoint='resnet101') net = net_cpu.cuda() decode = openpifpaf.decoder.factory_decode(net, seed_threshold=0.5) processor = openpifpaf.decoder.Processor(net, decode, instance_threshold=0.2, keypoint_threshold=0.3) data = openpifpaf.datasets.PilImageList([pil_im]) loader = torch.utils.data.DataLoader(data, batch_size=1, pin_memory=True) keypoint_painter = openpifpaf.show.KeypointPainter(color_connections=True, linewidth=6) for images_batch, _, __ in loader: images_batch = images_batch.cuda() fields_batch = processor.fields(images_batch) predictions = processor.annotations(fields_batch[0]) with openpifpaf.show.image_canvas(im) as ax: keypoint_painter.annotations(ax, predictions)