如何在 Google Colab 上運行 Samurai

我的原始貼文
https://baxin.netlify.app/how-to-run-samurai-on-google-colab/

什麼是武士？

SAMURAI：採用運動感知記憶體的零鏡頭視覺追蹤的分段任意模型

要求

• Google Colab 的 Google 帳號
• 抱臉帳號下載資料

如何在 Google Colab 上運行 Samurai

步驟 0. 取得 Hugging Face 令牌並將其新增至您的環境變數中

我們需要存取 Hugging Face 才能下載資料。

如果您不知道如何取得擁抱臉令牌，請參閱此頁面。
另外，如果您不知道如何將 Hugging Face 令牌新增至環境變數中，請查看這篇文章。

步驟 1. 更改預設運行時

要在 Google Colab 上運行 Samurai，我們需要將預設運行時更改為 GPU。
我們需要使用 T4（免費 GPU）。

步驟 2. 安裝軟體包

步驟 3. 複製 Samurai 儲存庫

步驟 4. 安裝 Sam2

步驟 5. 下載檢查點

步驟 6. 從 Hugging Face 下載數據

在這一部分中，我們將使用 python 腳本來設定 samurai repo 在資料準備部分中提到的資料。
https://github.com/yangchris11/samurai?tab=readme-ov-file#data-preparation

我們將使用的資料是l-lt/LaSOT

在本例中，我們將下載 cat 資料集，因此如果您想嘗試其他資料集，可以相應地變更程式碼。

步驟 7. 推理

最後一步是運行 Samurai 推理。
推理需要一段時間。

如果一切順利，您應該看到以下輸出：

所有程式碼都可以在此 GitHub 儲存庫中取得。

如果您喜歡這篇文章，請在 GitHub 上給它一個星星。

以上是如何在 Google Colab 上運行 Samurai的詳細內容。更多資訊請關注PHP中文網其他相關文章！