AI打遊戲-叄(標註圖片)

發表於2023-09-20

目標

  • 安裝標註工具
  • 標註遊戲圖片
  • 匯出到本地目錄當做訓練集

步驟

安裝標註工具

方式1:Docker

docker run --name label-studio -itd -p 8080:8080 heartexlabs/label-studio:latest

方式2:pip安裝

# 安裝
pip install -i https://mirrors.aliyun.com/pypi/simple/ label-studio

# 啟動
label-studio start

建立專案

  • 開啟LabelStudio,直接註冊一個賬號

    • 註冊使用郵箱,但是不校驗,隨便寫個郵箱格式就行
  • 登入後建立專案,建立專案有三個選項

    • Project Name: 專案名稱
    • Data Import: 上傳圖片資料

      • 上傳圖片有100張的限制,分幾次上傳幾次即可
    • Labeling Setup: 標註資訊設定

      • 選擇"Object Detection with Bounding Boxes"
      • 設定標籤,左側寫入"玩家/武士/地獄犬/螢火蟲",然後新增到右側
      • 右側的"Labels",刪除預設的標籤,點選標籤可以設定顏色(將四個標籤設定成差別較大的顏色)
  • 最後點選"Save"儲存

2_0_labelstudio-import.png
2_1_labelstudio-boxes.png
2_2_labelstudio-label.png

  • 點選剛才建立的專案,進入標註
  • 圖片下方是設定的標籤,點選後在畫面上拖動,將元素框選起來

    • 標籤邊上的數字是快捷鍵
  • 完成一張後點選右下方的"Submit"儲存標註資訊
  • 點選下一張繼續標註

2_3_labelstudio-labeling.png

匯出圖片及標註檔

  • 在專案的圖片列表中,點選右上角"Export"匯出資料
  • 格式選擇"YOLO",然後"Export"下載
  • 之後解壓即可

2_4_labelstudio-export.png

本節結果

  • 解壓後如下目錄結構
project-xxx
│
│  classes.txt  # 每行一個label,從0開始(這裡排序可能和設定不同)
│  notes.json   # label的json格式
│
├─images
│      58ffa2e4-357.jpg     # 所有圖片
│      aa23ddh4-245.jpg
│
└─labels
        58ffa2e4-357.txt    # 圖片對應的YOLO格式標註檔
        aa23ddh4-245.txt

相關文章