检查代码错误 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,dataformats='HWC') #y=2x for i in range(100): writer.add_scalars("y=2x",3*i,i) writer.close()

根据您提供的代码，我注意到您可能错过了一个缩进，这可能导致在循环结束之前关闭了TensorBoard的写入器。请尝试将writer.close()语句移动到循环结束后，以确保所有的记录都被正确地写入到TensorBoard日志文件中。

此外，您在使用writer.add_image()函数时指定了"dataformats"参数，但是根据PyTorch官方文档，此参数应该是"dataformat"，也就是说没有"s"。您可以尝试将"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()

如果问题仍然存在，请提供完整的错误信息和代码上下文，以便我更好地帮助您解决问题。