pytorch_Tensorboard的使用

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=‘CHW’，默认CHW，tensor是CHW，numpy是HWC
      '''
      # 在这行代码中加入参数 （dataformats=’HWC‘）
       
      writer.add_image()
      # y = x
      for i in range(100):
          writer.add_scalar("y = 2x", 2*i, i)
      
      
      writer.close()