Platform For AI：映像關鍵點訓練

前提條件

已開通OSS並完成授權，詳情請參見開通OSS服務和雲產品依賴與授權：Designer。

功能限制

• 僅Designer提供該演算法組件。

• 支援的計算引擎為DLC。

演算法簡介

關鍵點檢測模組提供主流的自上而下（Top-Down）的演算法，主要包含兩個部分：目標檢測和單人人體關鍵點檢測。人體關鍵點定位依賴於檢測演算法提出的Proposals，目前已經支援的模型包括HRNet和Lite-HRNet。

映像關鍵點訓練組件位於組件庫視覺演算法檔案夾下的離線訓練模型子檔案夾。

可視化配置組件

• 輸入樁

  輸入樁（從左至右）	限制資料類型	建議上遊組件	是否必選
  訓練資料	OSS	讀OSS資料	否
  訓練資料標註檔案	OSS	讀OSS資料	否
  驗證資料	OSS	讀OSS資料	否
  驗證資料標註檔案	OSS	讀OSS資料	否
  資料集標籤檔案	OSS	讀OSS資料	否

• 組件參數

  頁簽	參數	是否必選	描述	預設值
  欄位設定	訓練模型類型	是	訓練模型使用的演算法類型，僅支援TopDown。	TopDown
  	儲存訓練輸出的OSS目錄	否	儲存訓練模型的OSS目錄，比如：oss://examplebucket/output_dir/ckpt/。	無
  	訓練資料oss路徑	否	如果您沒有通過輸入樁配置該組件的訓練資料，則需要配置該參數。 訓練資料所在的OSS目錄。比如：oss://examplebucket/data/train_images/。 說明 如果您同時通過輸入樁和該參數配置了該組件的訓練資料，則優先使用輸入樁配置的資料。	無
  	訓練資料標註結果檔案路徑	否	如果您沒有通過輸入樁配置該組件的訓練資料標註結果檔案，則需要配置該參數。 訓練資料標註結果檔案所在的OSS路徑，比如：oss://examplebucket/data/annotations/train.json 。 說明 如果您同時通過輸入樁和該參數配置了該組件的訓練資料標註結果檔案，則優先使用輸入樁配置的資料。	無
  	驗證資料oss路徑	否	如果您沒有通過輸入樁配置該組件的驗證資料，則需要配置該參數。 驗證資料所在的OSS目錄，比如：oss://examplebucket/data/val_images/。 說明 如果您同時通過輸入樁和該參數配置了該組件的驗證資料，則優先使用輸入樁配置的資料。	無
  	驗證資料標註結果檔案路徑	否	如果您沒有通過輸入樁配置該組件的驗證資料標註結果檔案，則需要配置該參數。 驗證資料標註結果檔案所在的OSS路徑，比如：oss://examplebucket/data/annotations/val.json。 說明 如果您同時通過輸入樁和該參數配置了該組件的驗證資料標註結果檔案，則優先使用輸入樁配置的資料。	無
  	dataset info檔案oss路徑	是	如果您沒有通過輸入樁配置該組件的資料集標籤檔案，則需要配置該參數。 表示資料集標籤檔案所在的OSS路徑，比如：oss://examplebucke/data/annotations/dataset_info.py。 說明 如果您同時通過輸入樁和該參數配置了該組件的資料集標籤檔案，則優先使用輸入樁配置的資料。	無
  	資料來源格式	是	輸入資料的格式類型，僅支援DetSourceCOCO。	DetSourceCOCO
  	預訓練模型oss路徑	否	如果您有自己的預訓練模型，則將該參數配置為預訓練模型的OSS路徑。如果沒有配置該參數，則使用PAI提供的預設預訓練模型。	無
  參數設定	關鍵點檢測模型使用的backbone	是	選擇使用的骨幹模型，支援以下兩種主流模型： hrnet lite_hrnet	hrnet
  	關鍵點類別數目	是	資料中類別標籤的數目。	無
  	圖片resize大小	是	圖片大小調整成固定的高和寬，用半形逗號（,）分隔。	192,256
  	初始學習率	是	初始學習率大小。	0.01
  	學習率調整策略	是	使用學習率調整策略來控制學習率。支援的學習率調整策略為step：人工指定各階段的學習率。	step
  	Ir step	是	與學習率調整策略配合使用，多個step用半形逗號（,）串連。當epoch數量達到某個階段，學習率預設衰減0.1倍。 比如：初始學習率為0.1，總的訓練迭代epoch輪數為20，lr step為5,10。則epoch輪數為1~5時，學習率為0.1；epoch輪數為5~10時，學習率為0.01；epoch輪數為10~20時，學習率為0.001。	170,200
  	訓練batch_size	是	訓練的批大小，即模型訓練過程中，每次迭代（每一步）訓練的樣本數量。	32
  	評估batch_size	是	評估（驗證）的批大小，即模型驗證過程中，每次迭代（每一步）載入的樣本數量。	32
  	總的訓練迭代epoch輪數	是	所有樣本訓練完成一輪表示一個epoch。總的epoch輪數表示所有樣本共訓練多少輪。	200
  	儲存checkpoint的頻率	否	儲存模型檔案的頻率。取值為1表示1個epoch訓練完成後儲存一次模型。	1
  執行調優	最佳化方法	是	模型訓練的最佳化方法，支援以下取值： SGD Adam	SGD
  	每個GPU讀取訓練資料的進程數	否	每個GPU讀取訓練資料的進程數量。	2
  	evtorch model 開啟半精度	否	選中該參數，表示使用半精度進行模型訓練，用來降低記憶體佔用。	無
  	單機或分布式DLC	是	組件啟動並執行引擎，您可以結合實際情況選擇。系統支援以下計算引擎： 單機DLC 分布式DLC	單機DLC
  	worker個數	否	當運行引擎為分布式DLC時，需要配置該參數。 訓練過程中，並發的進程（worker）數量。	1
  	cpu機型選擇	否	當運行引擎為分布式DLC時，需要配置該參數。 選擇啟動並執行CPU規格。	16vCPU+64GB Mem-ecs.g6.4xlarge
  	gpu機型選擇	是	選擇啟動並執行GPU規格。	8vCPU+60GB Mem+1xp100-ecs.gn5-c8g1.2xlarge

• 輸出樁

  輸出樁	資料類型	下遊組件
  輸出模型	OSS路徑。該路徑是您在欄位設定頁簽的儲存訓練輸出的oss目錄參數配置的OSS路徑，訓練產生的模型儲存在該路徑下。	通用映像預測

使用樣本