pytorch把图像二值化的代码

以下是使用PyTorch将图像二值化的代码示例：

import torch
import torchvision.transforms as transforms
from PIL import Image

# 加载图像并转换为Tensor
img = Image.open('image.jpg')
img_tensor = transforms.ToTensor()(img)

# 将图像二值化
threshold = 0.5
binary_tensor = torch.where(img_tensor > threshold, torch.tensor(1.), torch.tensor(0.))

# 将二值化后的Tensor转换为图像并保存
binary_img = transforms.ToPILImage()(binary_tensor)
binary_img.save('binary_image.jpg')

在上面的示例中，我们首先使用PIL库加载图像，然后使用torchvision.transforms中的ToTensor()函数将图像转换为Tensor。接着，我们使用PyTorch的where()函数，根据我们设定的二值化阈值将Tensor中的像素值转换为0或1。最后，我们将二值化后的Tensor转换为图像，并使用PIL库将其保存为文件。