DeepLabV3 訓練自定義資料集例項(2)

2021-10-06 14:58:15 字數 2049 閱讀 1039

7.註冊資料集

在/models/research/deeplab/datasets 路徑的data_generator.py 第93行copy該結構體:

_mydata_information = datasetdescriptor(

splits_to_sizes=,

num_classes=2, #共2類

ignore_label=255, #忽略第255類

)

然後找到 _datasets_information ,加上mydata....一行

_datasets_information =
如下圖:

修改 deeplab\ utils\ train_utils.py,在210行處

exclude_list = ['global_step','logits'] #本來只有global_step ,現在加上 logits,表示不載入邏輯層的引數

修改在models/research/deeplab/train.py裡 165行左右

修改models/research/deeplab/input_preprocess.py  128行左右

3 自定義函式

用乙個自定義的猜數字函式來說明自定義函式怎麼寫。自定義函式 defgame min max import random number random.randint min,max 產生乙個需要猜的數字 while true guess float input 請在 d到 d之間才乙個數字 min,m...

Android自定義View訓練 1

前言 最近這段時間想根據網上部落格的案例,自己練習自定義view,僅僅作為練習用,還有很多不足之處,見諒 案例取自 先上傳原始碼 public class practiceview1 extends view public practiceview1 context context,attribut...

tensorflow 2 1 自定義訓練

常常會遇到自定義網路結構的情況,自定結構後往往會有多個輸入,或者還需要自定義loss或者accuracy函式,那麼keras的fit就無法使用了,因此需要自定義訓練步驟 下面則自定義一次batch的訓練步驟,包含了計算loss,accuracy和梯度下降。tensorflow2.0 主推eager模...