标签：human sum 保存 maps str lightweight image 图片
if label['img_paths']=='000000011791.jpg':
#if label['img_paths']=='000000537548.jpg':
sum=0
print('imgpeth',label['img_paths'])
image=image.transpose((2, 0, 1))
print('shape',image.shape)
os.mkdir('/home/ubuntu/lightw/lightweight-human-pose-estimation.pytorch-master/11/'+str(sum))
#plt.imsave('/home/ubuntu/lightw/lightweight-human-pose-estimation.pytorch-master/11/'+str(sum)+'/'+str(sum)+'.png', image)
torchvision.utils.save_image(torch.tensor(image.astype(np.float32)),'/home/ubuntu/lightw/lightweight-human-pose-estimation.pytorch-master/11/'+str(sum)+'/'+str(sum)+'.png')
print('keypoint_maps',keypoint_maps.shape)
print('paf_maps',paf_maps.shape)
for k in range(19):
torchvision.utils.save_image(torch.tensor(keypoint_maps[k,:,:].astype(np.float32)),'/home/ubuntu/lightw/lightweight-human-pose-estimation.pytorch-master/11/'+str(sum)+'/'+str(k)+'h'+'.png')
#im = Image.fromarray(keypoint_maps[0,k,:,:])
#im.save('/home/ubuntu/lightw/lightweight-human-pose-estimation.pytorch-master/11/'+str(sum)+'/'+str(k)+'h'+'.png')
for k in range(38):
torchvision.utils.save_image(torch.tensor(paf_maps[k,:,:].astype(np.float32)),'/home/ubuntu/lightw/lightweight-human-pose-estimation.pytorch-master/11/'+str(sum)+'/'+str(k)+'off'+'.png')
#im = Image.fromarray(keypoint_maps[0,k,:,:])
#im.save('/home/ubuntu/lightw/lightweight-human-pose-estimation.pytorch-master/11/'+str(sum)+'/'+str(k)+'h'+'.png')
print('全部保存')
标签：human,
sum,
保存,
maps,
str,
lightweight,
image,
图片
From： https://www.cnblogs.com/hahaah/p/17147530.html