1. 程式人生 > >u-net程式換上自己的資料集全黑的 解決辦法

u-net程式換上自己的資料集全黑的 解決辦法

做了一個月的u-net網路,剛開始前兩個星期很懵,因為總是跑出去全黑或者全灰的影象。看到網上好像沒有類似的教程,於是我總結一下我這一個月的亂撞南牆,希望給大家一些建議。

百度,調參,幾乎菜鳥能想到的方法我都用了,但是還是依然那個樣子。比如:調低學習率,增大epoch,把所有除以255的地方改為255.0,等等。

從這周開始,覺得可能是自己資料出了問題,因為自己的圖是三通道的彩色圖,首先把自己的圖片資料改為了灰度圖,雖然跑出來不再是全黑,出現了白色的邊框,但是我並不是分割邊框啊.......

我開始了繼續改引數的道路,然而要麼白邊要麼全黑。

然後我發現凡是跑成功了的幾乎都是圖片尺寸較小,於是我從網上下載了一波資料集(原圖以及只做好了的label)把圖片尺寸resize成了512*512,跑出來的圖開始有了效果,雖然還不滿意,但是通過修改引數一定可以達到預期效果。

換成了自己的資料集依然出現很多白點和黑點,總結了一下,我覺得要麼是我的目標太小,要麼是label製作方法不正確。

但是沒有跑出自己資料集的小夥伴可以試一下我上面的方法。