記一個在訓練模型過程中自己給自己挖的坑

  根據一個圖像拼接和融合的需求,訓練一個模型,輸入爲一組圖像,輸出爲一張圖像,輸入數據和ground truth的像素值都歸一化到[-1, 1]   我當時使用了UNet結構,卷積和反捲積都單獨封裝了一個函數,方便調用,在函數內部,卷積都會默認接一relu激活層   訓練結果出來後,發現內容基本都能和ground truth對應上,但是顏色很怪異,特別接近灰色,如下   然後尋找原因許久未果,陷
相關文章
相關標籤/搜索