pytorch_Tensorboard的使用

发布时间 2023-10-17 15:00:54作者: 翻斗花园小美Q
  1.  SummaryWriter()将什么写入文件中,如果不指定的话,就写入默认的

    1. 需要两个方法
      writer.add_image() 
      wruter.add_scalar()

       

    2. from torch.utils.tensorboard import SummaryWriter
      # 导入tensorboard,再导入SummaryWriter这个类
      
      writer = SummaryWriter("logs")
      
      # 利用OpenCV读取图片,获得numpy型图片数据
      '''
      ima_path = "路径名"
      from PIL import Image
      img = Image.open(image_path)
      print(type(img))
      (输出的是PIL型的图片,要转换成tensor的或者numpy的)
      利用numpy。array(),对PIL图片进行转换:
          首先导入numpy   import numpy as np
                         img_array = np.array(img)
                         print(type(img_array))
                         (可能会报错)
      '''
      # 在这行代码中加入参数 (dataformats=’HWC‘)
       
      writer.add_image()
      # y = x
      for i in range(100):
          writer.add_scalar("y = 2x", 2*i, i)
      
      
      writer.close()