Maskrcnn訓練RGB-D四通道圖片

網上的開源的Mask R-CNN代碼是以RGB圖像作爲默認輸入圖像的,如果需要處理4通道圖像,需要對源碼進行一定的修改,參考github以爲大神的解決方案,Github地址。 1、在子類中Config,設置IMAGE_CHANNEL_COUNT爲4 2、在Config中,將MEAN_PIXEL從3個值更改爲4個值,需要對數據集的深度圖求解均值。(平均像素值是根據你的訓練集數據求出) 3、將util
相關文章
相關標籤/搜索