Kornia:微分可能なコンピュータービジョンライブラリ

KorniaはバックエンドにPyTorchを使用したコンピュータービジョンライブラリで、古典的なコンピュータービジョンの処理を微分可能にして昨今のディープラーニングモデルに統合するためのPython向けライブラリ。

Kornia



Korniaは、PyTorch用の微分可能なコンピュータービジョンライブラリです。

Korniaは、コンピュータービジョンの一般的な問題を解決するための一連のルーチンと、微分可能なモジュールで構成されています。このパッケージのコアではメインのバックエンドとしてPyTorchを使用して効率性を高め、複雑な関数の勾配の定義・計算にReverse-Modeの自動微分を利用しています。




概要

このライブラリは既存のパッケージからインスピレーションを得ており、モデルの訓練時にニューラルネットワーク内に挿入できるオペレーター(画像変換、エピポーラ幾何学、深度推定、テンソル上で直接動作するフィルタ処理・エッジ検出などの低レベル画像処理)を備えたパッケージのサブセットで構成されています。

具体的には、Korniaは以下のコンポーネントで構成されるライブラリです。

コンポーネント
概要
kornia 強力なGPUサポートを備えた微分可能コンピュータービジョンライブラリ
kornia.augmentation GPUでdata augmentationモジュール
kornia.color 色空間を変換する一連のルーチン
kornia.contrib ユーザー貢献・実験的なオペレーターのコンパイル
kornia.enhance 正規化・強度変換モジュール
kornia.feature 特徴検出モジュール
kornia.filters 画像フィルタリング・エッジ検出モジュール
kornia.geometry 画像変換、3D線形代数、様々なカメラモデルを使用した変換などを行う
幾何学的コンピュータービジョンライブラリ
kornia.losses 様々なビジョンタスクを解決するための損失関数のスタック
kornia.morphology モルフォロジー演算モジュール
kornia.utils 画像からテンソルへのユーティリティとビジョン問題のための行列

ソースコードはGitHubにあるけど、pipで簡単にインストールできる。



https://arxiv.org/abs/1910.02190

関連記事

Unity ARKitプラグインサンプルのチュートリアルを読...

SONY製のニューラルネットワークライブラリ『NNabla』

Open Shading Language (OSL)

機械学習手法『Random Forest』

SDカードサイズのコンピューター『Intel Edison』

ROMOハッカソンに行ってきた

MPFB2:Blenderの人体モデリングアドオン

Regard3D:オープンソースのStructure fro...

Model View Controller

PyMC:Pythonのベイズ統計ライブラリ

WebGL開発に関する情報が充実してきている

NumSharp:C#で使えるNumPyライクな数値計算ライ...

R-CNN (Regions with CNN featur...

C++始めようと思うんだ

1枚の画像からマテリアルを作成できる無料ツール『Materi...

WordPressプラグインによるサイトマップの自動生成

定数

OpenCV 3.1のsfmモジュールを試す

Raspberry Pi 2のGPIOピン配置

CycleGAN:ドメイン関係を学習した画像変換

サンプルコードにも間違いはある?

手を動かしながら学ぶデータマイニング

MLDemos:機械学習について理解するための可視化ツール

Pix2Pix:CGANによる画像変換

スクレイピング

MeshLab:3Dオブジェクトの確認・変換に便利なフリーウ...

オープンソースの顔認識フレームワーク『OpenBR』

Unityの薄い本

Deep Learningとその他の機械学習手法の性能比較

Amazon Web ServicesでWordPress

Twitter APIのPythonラッパー『python-...

オープンソースのIT資産・ライセンス管理システム『Snipe...

全脳アーキテクチャ勉強会

Fast R-CNN:ディープラーニングによる一般物体検出手...

今年もSSII

Physics Forests:機械学習で流体シミュレーショ...

AndroidもopenGLも初心者さ (でもJavaは知っ...

LLM Visualization:大規模言語モデルの可視化

CGALDotNet:計算幾何学ライブラリ CGALのC#ラ...

OpenCV 3.1から追加されたSfMモジュール

OpenMesh:オープンソースの3Dメッシュデータライブラ...

ArUco:OpenCVベースのコンパクトなARライブラリ

コメント