Skip to content

4.ローカル環境で物体検出をしてみる

まかろに edited this page Dec 16, 2024 · 1 revision
  • GoogleColabで学習したモデルから得られた重みを利用して、ローカル環境で物体検出を再現する⇒推論
  • 学習したら、runs/detect/train/weights/last.ptをダウンロードする
  • last.pt:Google Drive ダウンロードリンク

  • ローカル環境はJupyterを使用

  • Google Colabと同じ感じで推論する

    // 重み付けファイルでトレーニング
    model = YOLO("last.pt")
    
    // 推論実行
    results = model.predict(source="検出したい画像")
  • 得られた結果