lightweight 保存图片

        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('全部保存')