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


スポンサーリンク

関連記事

OpenCV 3.3.0-RCでsfmモジュールをビルド

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

Pylearn2:ディープラーニングに対応したPythonの機械学習ライブラリ

Live CV:インタラクティブにComputer Visionコーディングができるツール

CNN Explainer:畳み込みニューラルネットワーク可視化ツール

書籍『イラストで学ぶ ディープラーニング』

Math Inspector:科学計算向けビジュアルプログラミングツール

Google製オープンソース機械学習ライブラリ『TensorFlow』のWindows版が公開された

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

WordPressのサーバ引っ越し方法を考える

OpenCVでiPhone6sのカメラをキャリブレーションする

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

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

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

Netron:機械学習モデルを可視化するツール

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

iOSデバイスのためのフィジカル・コンピューティングツールキット『konashi(こなし)』

TorchStudio:PyTorchのための統合開発環境とエコシステム

MythTV:Linuxでテレビの視聴・録画ができるオープンソースプロジェクト

UnityのGameObjectの向きをScriptで制御する

機械学習での「回帰」とは?

Quartus II

Leap MotionでMaya上のオブジェクトを操作できるプラグイン

COLMAP:オープンソースのSfM・MVSツール

OpenCV

SSD (Single Shot Multibox Detector):ディープラーニングによる一般...

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

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

Boost オープンソースライブラリ

WordPressのテーマを自作する

.NETで使えるTensorFlowライクなニューラルネットワークライブラリ『NeuralNetwo...

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

ZBrushのZScript入門

クラスの基本

チャットツール用bot開発フレームワーク『Hubot』

顔追跡による擬似3D表示『Dynamic Perspective』

Point Cloud Utils:Pythonで3D点群・Meshを扱うライブラリ

Unityで強化学習できる『Unity ML-Agents』

UnityでPoint Cloudを表示する方法

書籍『ROSプログラミング』

Raspberry PiでIoTごっこ

ROSの薄い本

コメント