注:有的时候在绘图的时候会出现,以下bug:
报错代码:
Traceback (most recent call last):
File "D:\PyCharm\PyCharm Community Edition 2023.3.4\plugins\python-ce\helpers\pydev\pydevconsole.py", line 364, in runcode
coro = func()
File "<input>", line 8, in <module>
File "D:\Anaconda\envs\Pytorch_0\lib\site-packages\torch\utils\tensorboard\writer.py", line 614, in add_image
image(tag, img_tensor, dataformats=dataformats), global_step, walltime
File "D:\Anaconda\envs\Pytorch_0\lib\site-packages\torch\utils\tensorboard\summary.py", line 441, in image
image = make_image(tensor, rescale=rescale)
File "D:\Anaconda\envs\Pytorch_0\lib\site-packages\torch\utils\tensorboard\summary.py", line 486, in make_image
image = image.resize((scaled_width, scaled_height), Image.ANTIALIAS)
AttributeError: module 'PIL.Image' has no attribute 'ANTIALIAS'
原因为:PIL版本过高,不合适。重新安装低版本的Pillow即可。如:
再次运行,即可成功运行,并生成图像。