Colab(Colaboratory)是一個由 Google 提供的雲端 Python 編輯器,它的最大特色是「免安裝、免設定」,只要有瀏覽器就能開始寫程式。
開啟 Google Colab 頁面,就會有一份現成的教學文件,鉅細靡遺地將初學者需要了解的都一一說明。這份文件就是一個 Colab 筆記本,它不只是教學內容,更是一個可以直接互動的程式環境。
這份文件不只是教你怎麼用 Colab,還包含許多範例與資源:
依照自己的需求,跳過不感興趣的部分,或深入探索特定主題。
最重要的,Colab 不受到電腦硬體等級的限制
Colab 的說明文件中,一開始就是在介紹 Gemini
在 Colab 頁面的右上方,也會看到 Gemini 的功能鍵
對 Colab & Gemini 有個基本的認識後,接著就來建立一個 Colab 檔案
檔案 --> 在雲端硬碟中建立新筆記本 (可以試著在 cell 裡面寫寫 code)
已經開始放空的人就直接召喚 Gemini 吧
寫 Prompt 的地方(將任務交給 Gemini)
送出後,Gemini 就會開始規劃
如果接受 Gemini 的建議,就按下接受並自動執行(或是逐步執行也可以);Colab 就會開始一一生成內容,全部執行需要一點時間
不接受的話,選擇取消,也可以繼續修改 Prompt
# 安裝 YOLOv8
!pip install ultralytics
# 連接 Google Drive 保存訓練結果
from google.colab import drive
drive.mount('/content/drive')
一定要記得整理成 YOLO 格式(txt 標註檔只存 bbox 與類別),推薦用 Roboflow 標註,貼心又直覺。順便附上 Python 匯入 Roboflow 專案的方式:
# Roboflow 套件安裝
!pip install roboflow
from roboflow import Roboflow
rf = Roboflow(api_key="在這裡貼上 API_KEY")
project = rf.workspace().project("taiwan-night-market")
dataset = project.version("1").download("yolov8")
from ultralytics import YOLO
# 指定資料集 yaml
model = YOLO('yolov8n.pt')
model.train(data='/content/dataset/data.yaml', epochs=3, imgsz=640)
epochs=3
是不夠的,但是數字愈高,要花的時間愈久,如果是讓 Gemini 幫忙生成程式碼的話,按下執行之前,記得調整一下(因為 Gemini 認為 epochs=100 是標準值)
pt 檔案在 Roboflow 也可以找到:
回到 Roboflow,就在 Version 裡面
pt 檔案是一種與 PyTorch(一個開源機器學習框架)相關的檔案格式,全稱為 PyTorch Model。pt 檔案包含模型的架構、權重(weights)以及訓練狀態(如優化器參數)。例如,訓練一個夜市小吃偵測模型後,Roboflow 或 Ultralytics YOLO 會生成 .pt 檔案,供後續推理或進一步訓練使用。
Ref.