Step 1. 下載資料

https://github.com/aws-educate-tw/aws-educate-sagemaker-diffusion-model-workshop.git

  1. 點擊 code 下的 Download ZIP 下載所有檔案

image.png

  1. (For Windows) 在檔案總管中解壓縮全部檔案

image.png

Step 2. 準備訓練資料

  1. 準備20張以上圖片(注意:只能用PNG或JPG)

  2. 在以下路徑找到 Step 1 下載的 JSON檔,路徑如下:

    aws-educate-sagemaker-diffusion-model-workshop-main\SageMaker\example_training_data\dataset_info.json

  3. 用記事本或 VS Code 開啟 JSON 檔,更改 prompt 中的訓練主角名稱後儲存

    內容範例:

    {
    	"instance_prompt": "a photo of Hello Kitty cat",
    	"class_prompt": "a photo of a cat"
    }
    

    (檔案需命名為 dataset_info.json)

    instance_prompt 代表這張圖片裡的主體是誰 "a photo of Hello Kitty" 代表模型要學「Hello Kitty 這個特定角色/概念」

    class_prompt 代表這個主體屬於甚麼類別 "a photo of a cat" 表示 Hello Kitty 被分為「貓」這個類別

    要把紅底的 Hello Kitty cat 和 cat 改成自己準備的角色名稱跟那個角色所屬的類別,其他部分文字不用更改