deeplab v2語義分割 caffe配置

2021-08-19 18:37:03 字數 1394 閱讀 3657

我的硬體配置ubuntu16.04,gpu 1080ti,cuda8.0,cudnn5.0

caffe_bin=$/.build_release/tools/caffe.bin

#caffe_bin=$/build/tools/caffe

exp=voc12   //這一句很重要,按照一般設定,許多都設定為exp=.,這樣的話後面容易出現測試結果全為黑的問題,知乎上有人回答,但是我因為這個問題煩惱了好久,並且deeplab_v2網上資料確實不多。

if [ "$" = "voc12" ]; then

num_labels=21

data_root=$/voc_aug/dataset

else

num_labels=0

echo "wrong exp name"

fi## specify which model to train

########### voc12 ################

net_id=deeplab_largefov

## variables used for weakly or semi-supervisedly training

#train_set_suffix=

train_set_suffix=_aug     

#train_set_strong=train

#train_set_strong=train200

#train_set_strong=train500

#train_set_strong=train1000

#train_set_strong=train750

#train_set_weak_len=5000

dev_id=0

#####

## create dirs

config_dir=$/config/$

model_dir=$/model/$

mkdir -p $

log_dir=$/log/$

mkdir -p $

export glog_log_dir=$

## run

run_train=1     //要幹什麼,就讓哪個為1即可,一般後面的train2和test2用不到。

run_test=0

run_train2=0

run_test2=0

最後兩個,如圖:

5、如果要單張測試,可以修改voc12/list目錄下的val.txt和val_id.txt,只留下你要測試的那一行即可。

6、可以儲存訓練和測試時的日誌,比如訓練時,sh run_pascal.sh 2>&1|tee train.log。

deeplab v2 安裝問題總結

deeplabv2安裝問題總結如下 1.cudnn從v5降級到v4,版本5在我的系統出現bug 2.atomicadd的重寫問題,cuda8已經有了atomicadd函式的定義,出現bug。解決方法 具體做法 修改common.cuh ifndef caffe common cuh define c...

語義分割之deeplab v1

首先我們簡單考慮一下什麼是語義分割?語義分割是從粗推理到精推理的自然步驟。原點可以定位在分類,分類包括對整個輸入進行 下一步是本地化 檢測,它不僅提供類,還提供關於這些類的空間位置的附加資訊。最後,語義分割通過對每個畫素進行密集的 推斷標籤來實現細粒度的推理,從而使每個畫素都被標記為其封閉物件區域的...

語義分割deeplabv3原理與流程梳理

deeplabv3是一種語義分割網路,語義分割旨在對給定的每乙個畫素點進行類別 在這裡我們來梳理一下deeplabv3網路的大致流程僅供參考,參考的演算法實現位址為 演算法使用的資料集是分割常用的cityscapes,在對資料集進行適當的預處理後,輸入網路的是batchx3x256x256的和bat...