問題解決:writer.add_image添加的圖片模糊，很多點點；原來是要uint8

保存圖片:uint8類型轉換 訓練過程中用writer.add_image向tensorboardＸ添加圖片，發現圖片總是下面這種鬼樣子: 但是在前一行用import matplotlib.pyplot as plt保存的圖片就沒有問題。 經過查閱writer.add_image的源碼，發現其要求輸入的圖片數據類型如下: The elements in img_tensor can either

>>阅读原文<<