前提條件

已開通OSS并完成授權(quán)，詳情請(qǐng)參見開通OSS服務(wù)和云產(chǎn)品依賴與授權(quán)：Designer。

使用限制

• 僅Designer提供該算法組件。

• 僅支持DLC引擎。

可視化配置組件參數(shù)

• 輸入樁

  輸入樁（從左到右）	限制數(shù)據(jù)類型	建議上游組件	是否必選
  訓(xùn)練數(shù)據(jù)	OSS	讀OSS數(shù)據(jù)	否。 如果沒有通過輸入樁配置該參數(shù)，也可以在字段設(shè)置頁(yè)簽的訓(xùn)練數(shù)據(jù)oss路徑參數(shù)中配置。
  評(píng)估數(shù)據(jù)	OSS	讀OSS數(shù)據(jù)	否。 如果沒有通過輸入樁配置該參數(shù)，也可以在字段設(shè)置頁(yè)簽的測(cè)試數(shù)據(jù)oss路徑參數(shù)中配置。
  類別列表文件	OSS	讀OSS數(shù)據(jù)	否。 如果沒有通過輸入樁配置該參數(shù)，也可以在字段設(shè)置頁(yè)簽的類別列表文件oss路徑參數(shù)中配置。

• 組件參數(shù)

  頁(yè)簽	參數(shù)	是否必選	描述	默認(rèn)值
  字段設(shè)置	訓(xùn)練模型類型	是	訓(xùn)練所選的模型類型，支持以下取值： FCOS YOLOX YOLOv5 YOLOv7	YOLOX
  	訓(xùn)練所用oss目錄	否	存儲(chǔ)訓(xùn)練模型的OSS目錄。例如examplebucket.oss-cn-shanghai-internal.aliyuncs.com/test/ckpt/。假如為空，使用工作空間默認(rèn)路徑	無(wú)
  	訓(xùn)練集標(biāo)注結(jié)果文件路徑	否	如果數(shù)據(jù)源格式是PAI標(biāo)注格式，則提供文件后綴名為manifest的文件。 如果數(shù)據(jù)源格式是COCO格式，則提供后綴名為.json的文件。 如果通過了輸入樁配置算法組件的標(biāo)注訓(xùn)練數(shù)據(jù)，則為空。如果同時(shí)通過輸入樁和該參數(shù)配置了算法組件的標(biāo)注訓(xùn)練數(shù)據(jù)，則優(yōu)先使用輸入樁配置的輸入。	無(wú)
  	測(cè)試集標(biāo)注結(jié)果文件路徑	否	如果數(shù)據(jù)源格式是PAI標(biāo)注格式，則提供文件后綴名為manifest的文件。 如果數(shù)據(jù)源格式是COCO格式，則提供后綴名為.json的文件。 如果通過了輸入樁配置算法組件的標(biāo)注測(cè)試數(shù)據(jù)，則為空，如果同時(shí)通過輸入樁和該參數(shù)配置了算法組件的標(biāo)注測(cè)試數(shù)據(jù)，則優(yōu)先使用輸入樁配置的輸入。	無(wú)
  	類別列表文件oss路徑	否	填入后綴名為txt的標(biāo)簽列表文件。 如果通過了輸入樁配置算法組件的類別列表文件，則為空。如果同時(shí)通過輸入樁和該參數(shù)配置了算法組件的類別列表文件，則優(yōu)先使用輸入樁配置的輸入。	無(wú)
  	預(yù)訓(xùn)練模型oss路徑	否	如果有自己的預(yù)訓(xùn)練模型，則將該參數(shù)配置為自己預(yù)訓(xùn)練模型的OSS路徑。如果沒有配置該參數(shù)，則使用PAI提供的默認(rèn)預(yù)訓(xùn)練模型。	無(wú)
  	訓(xùn)練數(shù)據(jù)oss路徑	是	僅當(dāng)數(shù)據(jù)源格式為COCO格式時(shí)候，才會(huì)顯示填寫此單元格，此處填寫的為COCO格式數(shù)據(jù)的訓(xùn)練圖片路徑。	無(wú)
  	測(cè)試數(shù)據(jù)oss路徑	是	僅當(dāng)數(shù)據(jù)源格式為COCO格式時(shí)候，才會(huì)顯示填寫此單元格，此處填寫的為COCO格式數(shù)據(jù)的測(cè)試圖片路徑。
  	數(shù)據(jù)源格式	是	支持以下取值： COCO格式 PAI標(biāo)注格式（模型類型YOLOv5、YOLOv7僅支持PAI標(biāo)注格式的數(shù)據(jù)集）	PAI標(biāo)注格式
  參數(shù)設(shè)置	YOLOX模型結(jié)構(gòu)	是	僅當(dāng)訓(xùn)練模型類型為YOLOX時(shí)，才會(huì)顯示填寫此單元格，此處下拉選擇你所想選擇的結(jié)構(gòu)。 yolox-s yolox-m yolox-l yolox-x	yolo-s
  	檢測(cè)類別數(shù)目	是	數(shù)據(jù)中類別標(biāo)簽的數(shù)目。	20
  	圖像尺度	是	圖像resize后的大小，填寫時(shí)候記得空格：高和寬之間使用空格分隔例如 320 320。	320 320
  	優(yōu)化方法	是	僅訓(xùn)練模型類型為YOLOX時(shí)，支持配置該參數(shù)。 模型訓(xùn)練的優(yōu)化方法，支持以下取值： momentum adam	momentum
  	初始學(xué)習(xí)率	是	初始學(xué)習(xí)率	0.01
  	訓(xùn)練batch_size	是	訓(xùn)練的批大小，即單次模型迭代或訓(xùn)練過程中使用的樣本數(shù)量。	8
  	評(píng)估batch_size	是	評(píng)估的批大小，即單次模型迭代或訓(xùn)練過程中使用的樣本數(shù)量。	8
  	epoch輪數(shù)	是	總的訓(xùn)練迭代輪數(shù)。 說(shuō)明 如果是YOLOX，總的epoch需要大于warmup epochs + 最后穩(wěn)定lr的epochs。	20
  	loss打印頻率	否	loss（損失值）打印頻率。默認(rèn)為200，表示每200個(gè)訓(xùn)練批次，打印一次loss（損失值）。
  	驗(yàn)證頻率	否	默認(rèn)為2，表示相隔2個(gè)epoch進(jìn)行驗(yàn)證。
  	warmup epochs	否	僅當(dāng)模型為YOLOX時(shí)，需要填寫。	5
  	最后穩(wěn)定lr的epochs	否	僅當(dāng)模型為YOLOX時(shí)，需要填寫。	5
  	導(dǎo)出的模型類型	是	導(dǎo)出模型的格式，提供兩種導(dǎo)出格式。 raw jit（模型類型YOLOv5、YOLOv7不支持導(dǎo)出jit格式） onnx	raw
  	保存checkpoint的頻率	否	保存模型文件的頻率。取值1表示對(duì)所有訓(xùn)練數(shù)據(jù)都進(jìn)行一次迭代。	1
  執(zhí)行調(diào)優(yōu)	gpu機(jī)型選擇	是	此算法不支持沒有GPU的機(jī)器運(yùn)行。	4
  	開啟半精度f(wàn)p16	否	是否使用fp16半精度模式。	false
  	單機(jī)或分布式	是	只支持DLC分布方式訓(xùn)練。	分布式DLC

• 輸出樁

  輸出樁（從左到右）	數(shù)據(jù)類型	下游組件
  輸出模型	OSS路徑。該路徑是您在字段設(shè)置頁(yè)簽的訓(xùn)練所用oss目錄參數(shù)配置的OSS路徑，訓(xùn)練生成SavedModel格式的模型存儲(chǔ)在該路徑下。	通用圖像預(yù)測(cè)

示例