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

関連記事

網元AMIで作ったWordpressサイトのインスタンスをt...

Dlib:C++の機械学習ライブラリ

WordPress on Windows Azure

写真から3Dメッシュの生成・編集ができる無料ツール『Auto...

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

FCN (Fully Convolutional Netwo...

Web経由でRaspberry PiのGPIOを操作したい

オープンソースの物理ベースレンダラ『Mitsuba』をMay...

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

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

ZBrushで作った3Dモデルを立体視で確認できるVRアプリ...

ManimML:機械学習の概念を視覚的に説明するためのライブ...

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

今年もSSII

オープンソースの物理ベースGIレンダラ『appleseed』

Adobeの手振れ補正機能『ワープスタビライザー』の秘密

クラスの基本

組み込み向けのWindows OS 『Windows Emb...

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

Raspberry Pi 2のGPIOピン配置

Adobe MAX 2015

AnacondaとTensorFlowをインストールしてVi...

OpenGV:画像からカメラの3次元位置・姿勢を推定するライ...

UnityからROSを利用できる『ROS#』

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

AmazonEC2のインスタンスをt1.microからt2....

機械学習で遊ぶ

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

OpenCVでカメラ画像から自己位置認識 (Visual O...

OpenCVでPhotoshopのプラグイン開発

OpenCVで平均顔を作るチュートリアル

CGレンダラ研究開発のためのフレームワーク『Lightmet...

PythonでMayaのShapeノードプラグインを作る

Point Cloud Libraryに動画フォーマットが追...

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

OpenGVの用語

UnityでOpenCVを使うには?

顔検出・認識のAPI・ライブラリ・ソフトウェアのリスト

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

オープンソースのプリント基板設計ツール『KiCad』

顔画像処理技術の過去の研究

頭蓋骨からの顔復元と進化過程の可視化

コメント