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で画面タッチ・ジェスチャ入力を扱う無料Asset『...

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

Mitsuba 2:オープンソースの物理ベースレンダラ

Pythonの自然言語処理ライブラリ『NLTK(Natura...

ベイズ推定とグラフィカルモデル

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

iPhoneアプリ開発 Xcode 5のお作法

pythonもかじってみようかと

続・ディープラーニングの資料

OpenFace:Deep Neural Networkによ...

Multi-View Environment:複数画像から3...

MVStudio:オープンソースのPhotogrammetr...

Raspberry PiでIoTごっこ

フォトンの放射から格納までを可視化した動画

Structure from Motion (多視点画像から...

Unity MonoBehaviourクラスのオーバーライド...

YOLO (You Only Look Once):ディープ...

FCN (Fully Convolutional Netwo...

オープンソースの人体モデリングツール『MakeHuman』の...

OpenSfM:PythonのStructure from ...

UnityでOpenCVを使うには?

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

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

Kinect for Windows v2の日本価格決定

TeleSculptor:空撮動画からPhotogramme...

libigl:軽量なジオメトリ処理ライブラリ

OpenCVの三角測量関数『cv::triangulatep...

TensorSpace.js:ニューラルネットワークの構造を...

WordPress on Windows Azure

KelpNet:C#で使える可読性重視のディープラーニングラ...

ZScript

C++ 標準テンプレートライブラリ (STL)

ブログが1日ダウンしてました

Konashiを買った

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

OpenGVの用語

書籍『3次元コンピュータビジョン計算ハンドブック』を購入

3Dグラフィックスの入門書

Google App Engineのデプロイ失敗

C#で使える遺伝的アルゴリズムライブラリ『GeneticSh...

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

CGのためのディープラーニング

コメント