人体关键点检测

人体骨骼关键点检测主要检测人体的一些关键点，如关节、五官等，通过关键点描述人体骨骼信息。MS COCO数据集是多人人体关键点检测数据集，具有关键点个数为17，图像的样本数多于30万张，也是目前的相关研究中最常用的数据集。在torchvision库中，提供了已经在MS COCO数据集上与训练的keypointcnn_resnet50_fpn()网络模型，该网络可以用于人体的关键点检测。先导入与训练好的网络模型，程序如下所示：

import numpy as np
import torchvision
import torch
import torchvision.transforms as transforms
from PIL import  Image,ImageDraw
import matplotlib.pyplot as plt
import matplotlib.image as mping
model = torchvision.models.detection.keypointrcnn_resnet50_fpn(pretrained=True)
model.eval()
print(model)

输出结果如下：