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

保存圖片:uint8類型轉換

訓練過程中用writer.add_image向tensorboardX添加圖片,發現圖片總是下面這種鬼樣子:
在這裏插入圖片描述但是在前一行用import matplotlib.pyplot as plt保存的圖片就沒有問題。
經過查閱writer.add_image的源碼,發現其要求輸入的圖片數據類型如下:
The elements in img_tensor can either have values in [0, 1] (float32) or [0, 255] (uint8)
於是將圖片利用np.array(img, dtype=‘uint8’)轉換爲uint8類型。之後發現tensorboard上變得正常了。
在這裏插入圖片描述

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章