编程白! 导入torchvision报错,但是版本和torch对应一致
from PIL import Image
from torch.utils.tensorboard import SummaryWriter
from torchvision import transforms
writer = SummaryWriter("logs")
img = Image.open("hymenoptera_data/train/ants/0013035.jpg")
#Totensor
trans_totensor = transforms.ToTensor()
img_tensor = trans_totensor(img)
writer.add_image("totensor",img_tensor)
writer.close()
UserWarning: Failed to load image Python extension:
warn(f"Failed to load image Python extension: {e}")