TypeError: add_image() got an unexpected keyword argument 'dataformat'
时间: 2023-07-31 09:07:38 浏览: 255
提示错误的图片
4星 · 用户满意度95%
抱歉,我之前犯了一个错误。在PyTorch 1.5.0版本之前,add_image()函数确实接受"dataformats"参数,但是在1.5.0版本之后,参数名称已更改为"dataformat",并且"dataformats"参数已被弃用。
因此,您需要将"dataformats"参数改为"dataformat"。修改后的代码如下所示:
```
from torch.utils.tensorboard import SummaryWriter
import numpy as np
from PIL import Image
writer = SummaryWriter("logs")
image_path = "D:\\ymproject\\learn pytorch\\DataSet\\train\\ants_image\\0013035.jpg"
img_PIL = Image.open(image_path)
img_array = np.array(img_PIL)
print(type(img_array))
print(img_array.shape)
writer.add_image("test", img_array, 1, dataformat='HWC')
# y=2x
for i in range(100):
writer.add_scalars("y=2x", 3*i, i)
writer.close()
```
希望这次能够帮助您解决问题。如果仍然有任何疑问,请随时告诉我。
阅读全文