単眼深度推定と焦点ずれ推定は、コンピューター ビジョンにおける 2 つの基本的なタスクです。ほとんどの既存の方法は、深度推定と焦点ぼけ推定を 2 つの別個のタスクとして扱い、それらの間の強い関連性を無視しています。この作業では、2 つのデコーダーを備えたエンコーダーで構成されるマルチタスク学習ネットワークを提案し、1 つの焦点を合わせた画像から深度とデフォーカス マップを推定します。マルチタスク ネットワークを介して、深度推定は、弱いテクスチャ領域でより良い結果を得るために焦点ぼけ推定を容易にし、焦点ぼけ推定は、2 つのマップ間の強力な物理的接続によって深度推定を容易にします。データセット (ALL-in-3D データセットという名前) をセットアップしました。これは、10 万セットの全焦点画像、焦点深度のある焦点画像、深度マップ、デフォーカス マップで構成される最初の完全実画像データセットです。これにより、ネットワークは深度画像と実際のデフォーカス画像の間の特徴と強固な物理的接続を学習できます。実験は、ネットワークが合成の焦点を合わせた画像よりも実際に焦点を合わせた画像からより多くの固体の特徴を学習することを示しています。さまざまなタスクが互いに促進するこのマルチタスク構造の恩恵を受けて、深度と焦点ぼけの推定は、他の最先端のアルゴリズムよりも大幅に優れたパフォーマンスを実現します。コードとデータセットは、https://github.com/cubhe/MDDNet で公開されます。
Monocular depth estimation and defocus estimation are two fundamental tasks in computer vision. Most existing methods treat depth estimation and defocus estimation as two separate tasks, ignoring the strong connection between them. In this work, we propose a multi-task learning network consisting of an encoder with two decoders to estimate the depth and defocus map from a single focused image. Through the multi-task network, the depth estimation facilitates the defocus estimation to get better results in the weak texture region and the defocus estimation facilitates the depth estimation by the strong physical connection between the two maps. We set up a dataset (named ALL-in-3D dataset) which is the first all-real image dataset consisting of 100K sets of all-in-focus images, focused images with focus depth, depth maps, and defocus maps. It enables the network to learn features and solid physical connections between the depth and real defocus images. Experiments demonstrate that the network learns more solid features from the real focused images than the synthetic focused images. Benefiting from this multi-task structure where different tasks facilitate each other, our depth and defocus estimations achieve significantly better performance than other state-of-art algorithms. The code and dataset will be publicly available at https://github.com/cubhe/MDDNet.