Kubric:機械学習用アノテーション付き動画生成パイプライン

久しぶりにコンピュータビジョン系の話題。

Google Researchから機械学習用のアノテーション付き動画を生成するためのツールがオープンソースで公開された↓

Kubric



Kubricは、Instance SegmentationマスクやDepthマップ、オプティカルフローなどのリッチなアノテーション付きのセミリアルな動画を作成するためのデータ生成パイプラインです。

 ※このプロジェクトはまだアルファ段階であり、大幅に変更される可能性があります。

モチベーションと設計

機械学習システムの訓練と評価、特にunsupervised multi-object video understandingにおいてはより良いデータが必要です。既存のシステムは、toy datasetsでは上手く行くものの、現実世界のデータでは失敗してしまいます。複雑さの異なる適切なデータセットをオンデマンドで作成できれば、進歩を大幅に加速できる可能性があります。
Kubricは、主にpybullet(物理シミュレーション用)とBlender(レンダリング用)をベースに構築されていますが、コードはモジュール式に保たれているため、様々なレンダリングバックエンドをサポートできる可能性があります。

Kubricは特に数千台のマシンを使って処理できるスケーラビリティに重点を置いて開発されているらしい。
詳しくはarXivで公開されているKubric: A scalable dataset generatorや、公式ドキュメントを参照。



レンダリングに使われているBlenderについてはもはや説明不要でしょう。
物理シミュレーションに使われているpybulletというのは、Bullet Physics SDKというC++の物理シミュレーションフレームワークのPythonバインディング。強化学習方面ではすでに結構利用されているらしい。
https://zenn.dev/ymd_h/articles/14397e6ae7ea3d

Kubricの公式ドキュメントを読むと、一式揃ったDockerイメージも用意されているようですね。
https://kubric.readthedocs.io/en/latest/source/0_installing.html

ここ数年、機械学習用の画像・動画データをCGで生成する需要に応えるツールが少しずつ増えてきたな。


BlenderProc:Blenderで機械学習用の画像データを生成するPythonツール
機械学習用の画像データをCGで生成するのもData Augmentation(データ拡張)の範疇でしょうか。物体や風景画像だけでなく、Depthや物体ラベル、Semantic Segmentation用のアノテーション画像も必要となると、C...


最近はUnityが公式にUnity Computer Visionというデータセット生成サービスを始めていて、それに使用しているPerception Packageも公開している↓

Perception Package (Unity Computer Vision)



Perceptionパッケージは、コンピュータビジョンの訓練と検証のための大規模なデータセットを生成するためのツールキットを提供します。 現在は、カメラベースのユースケースに焦点を当てており、最終的には他の形式のセンサーや機械学習タスクに拡張される予定です。

ツールと製品の詳細についてはUnity Computer Visionページにアクセスしてください。

https://blog.unity.com/ja/technology/supercharge-your-computer-vision-models-with-synthetic-datasets-built-by-unity

関連記事

今年もSSII

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

オープンソースのネットワーク可視化ソフトウェアプラットフォーム『Cytoscape』

Vancouver Film Schoolのデモリール

WordPress on Windows Azure

ZBrushで仮面ライダー3号を造る 仮面編 横顔のシルエットをリファレンスに合わせる

BlenderのPython環境にPyTorchをインストールする

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

UnityでTweenアニメーションを実装できる3種類の無料Asset

ZBrushでアヴァン・ガメラを作ってみる 全体のバランス調整

ヘッドマウントディスプレイとビジュアリゼーションの未来

C++の抽象クラス

ZBrushで仮面ライダー3号を造る 仮面編 PolyGroupを分割する

実写と実写の合成時の色の馴染ませテクニック

ZBrushでアヴァン・ガメラを作ってみる 頬の突起を作り始める

PGGAN:段階的に解像度を上げて学習を進めるGAN

FacebookがDeep learningツールの一部をオープンソース化

PythonでBlenderのAdd-on開発

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

レンダラ制作はOpenGL とか DirectX を使わなくてもできるんだぜ

書籍『仕事ではじめる機械学習』を読みました

Pythonのソースコードに特化した検索エンジン『Nullege』

Houdiniのライセンスの種類

トランスフォーマーの変形パターンを覚えるのは楽しいと思うんだ

立体視を試してみた

FCN (Fully Convolutional Network):ディープラーニングによるSema...

Unreal Engine Tokyo MeetUp!へ行ってきた

konashiのサンプルコードを動かしてみた

Mayaのポリゴン分割ツールの進化

Webスクレイピングの勉強会に行ってきた

ZBrushトレーニング

Blender 2.8がついに正式リリース!

中学3年生が制作した短編映像作品『2045』

SSII 2014 デモンストレーションセッションのダイジェスト動画

AfterEffectsプラグイン開発

Iterator

CGレンダラ研究開発のためのフレームワーク『Lightmetrica (ライトメトリカ)』

3DCGのモデルを立体化するサービス

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

『ゴジラ キング・オブ・モンスターズ』のVFXブレイクダウン

iPadをハンディ3Dスキャナにするガジェット『iSense 3D Scanner』

ジュラシック・パークのメイキング

コメント