Repolode
albumentations-team

albumentations-team/AlbumentationsX

コンピュータビジョン向けの画像拡張Pythonライブラリ。ピクセル単位、空間、医療画像、3Dの変換に対応し、マスク、ボックス、キーポイントを一貫して扱えます。

スター
572
フォーク
43
未解決の Issue
33
最終コミット
一昨日

これは何?

AlbumentationsXは、既存の画像から新しい学習サンプルを生成し、ディープラーニングモデルの性能向上を図る画像拡張のためのPythonライブラリです。画像に加えてマスク、バウンディングボックス、キーポイントにも同じランダム変換を適用する統一的なパイプラインAPIを提供し、アノテーションの位置関係を保ちます。

ぼかし、ノイズ、色調の変更、天候エフェクトなど70種類以上のピクセル単位の変換に加え、クロップ、反転、弾性変形、モザイクなどの空間変換を利用できます。バイアスフィールドやギブスリンギングといった医療画像向け変換や、ボリュームデータ向けの3D変換にも対応しています。PyTorchやTensorFlowのパイプラインとの連携例も紹介されています。

このプロジェクトはPyPIで配布されており、Python 3.11以上が必要です。ライセンスはAGPL-3.0-onlyで、メンテナーから商用ライセンスも入手できます。

主な機能

  • 01統一APIにより、画像、マスク、バウンディングボックス、キーポイントにまとめて変換を適用できます。
  • 02ぼかし、ノイズ、色調整など、70種類を超えるピクセル単位の画像拡張に対応しています。
  • 03クロップ、反転、アフィン変換、弾性変形、モザイクなどの空間変換を利用できます。
  • 04医療画像向けの変換と、ボリュームデータ向けの3D変換に対応しています。
  • 05PyTorchやTensorFlowの学習パイプラインで利用できます。

ユースケース

  • 物体検出やセグメンテーションの学習に使う、ラベル付きデータセットを拡充する。
  • マスクの位置関係を保ちながら、医療画像やボリュームデータを拡張する。
  • 画像分類モデル向けに、再現性のある前処理パイプラインを構築する。

クイックスタート

  1. 1

    お使いのプラットフォーム向けのPyTorchビルドをインストールします(LinuxのCPU専用版を例として表示)

    pip install "torch>=2.13.0" --index-url https://download.pytorch.org/whl/cpu
  2. 2

    OpenCVとともにAlbumentationsXをインストールします(headless版)

    pip install "albumentationsx[headless]"
  3. 3

    ライブラリをインポートし、変換パイプラインを定義します

使う前に知っておきたいこと

  • AGPL-3.0-onlyライセンスで提供されています。コピーレフトライセンスであり、ネットワーク経由での利用や派生物にはソースコードの共有義務があります。
  • Python 3.11以降が必要です。
  • ライブラリのインポートには、別途インストールしたPyTorchビルドが必要です。CPU、CUDA、MPSの環境に合わせて選択してください。

トピック

よくある質問

AlbumentationsX とは?
コンピュータビジョン向けの画像拡張Pythonライブラリ。ピクセル単位、空間、医療画像、3Dの変換に対応し、マスク、ボックス、キーポイントを一貫して扱えます。 AlbumentationsXは、既存の画像から新しい学習サンプルを生成し、ディープラーニングモデルの性能向上を図る画像拡張のためのPythonライブラリです。画像に加えてマスク、バウンディングボックス、キーポイントにも同じランダム変換を適用する統一的なパイプラインAPIを提供し、アノテーションの位置関係を保ちます。
AlbumentationsX は無料で使えますか?ライセンスは?
AlbumentationsX は AGPL-3.0 ライセンスのオープンソースです。商用製品で使う前にライセンス条件を確認してください。
AlbumentationsX は今もメンテナンスされていますか?
活発にメンテナンス中。AlbumentationsX の最終コミットは 2026年10月19日 です。
AlbumentationsX は何に使えますか?
物体検出やセグメンテーションの学習に使う、ラベル付きデータセットを拡充する。 マスクの位置関係を保ちながら、医療画像やボリュームデータを拡張する。 画像分類モデル向けに、再現性のある前処理パイプラインを構築する。
AlbumentationsX のインストール方法は?
お使いのプラットフォーム向けのPyTorchビルドをインストールします(LinuxのCPU専用版を例として表示): pip install "torch>=2.13.0" --index-url https://download.pytorch.org/whl/cpu OpenCVとともにAlbumentationsXをインストールします(headless版): pip install "albumentationsx[headless]" ライブラリをインポートし、変換パイプラインを定義します
AlbumentationsX はどこで入手できますか?
ソースコードとインストール手順は GitHub にあります:https://github.com/albumentations-team/AlbumentationsX

概要はプロジェクトの README をもとに AI が生成しています。GitHub の数値は毎日更新されます。

関連プロジェクト