? RGB-D-Fusion: Difusi Kedalaman yang Dikondisikan Gambar dari Subjek Humanoid

Penulis:
Sascha Kirch, Valeria Olyunina, Jan Ondřej, Rafael Pages, Sergio Martín & Clara Pérez-Molina

[ Paper ] [ BibTex ]

Implementasi TensorFlow untuk RGB-D-Fusion. Untuk detailnya, lihat makalah RGB-D-Fusion: Difusi Kedalaman Berkondisi Gambar pada Subjek Humanoid .

Kontribusi

Kami menyediakan kerangka kerja untuk estimasi kedalaman monokuler padat resolusi tinggi menggunakan model difusi .
Kami melakukan resolusi super untuk data kedalaman padat yang dikondisikan pada kondisi input RGB-D multi-modal menggunakan model difusi.
Kami memperkenalkan teknik augmentasi baru , yaitu noise kedalaman, untuk meningkatkan kekokohan model resolusi super kedalaman.
Kami melakukan ablasi dan eksperimen yang ketat untuk memvalidasi pilihan desain kami

Berita

2023/10/14 : Kode tersedia Sekarang!
2023/09/04 : Makalah kami sekarang diterbitkan di IEEE Access!
2023/07/29 : Kami merilis pra-cetak kami di arxiv.

Kerangka

kerangka fusi rgb-d

?️ Hasil

Prediksi vs. GT

hasil_gt

Dalam prediksi liar

Instalasi

Kami merekomendasikan menggunakan lingkungan buruh pelabuhan. Kami menyediakan file buruh pelabuhan dari TensorFlow dan file buruh pelabuhan dari nvidia. Yang terakhir lebih besar tetapi mencakup optimalisasi kinerja nvidia. Pastikan buruh pelabuhan diinstal termasuk ekstensi GPU nvidia.

Bangun gambarnya

docker build -t < IMAGE_NAME > / < VERSION > -f < PATH_TO_DOCKERFILE >

Buat wadahnya

docker container create --gpus all -u 1000:1000 --name rgb-d-fusion -p 8888:8888 -v < PATH_TO_tf_DIR > :/tf -v < PATH_TO_YOUR_GIT_DIR > :/tf/GitHub -it < IMAGE_NAME > / < VERSION >

Mulai wadahnya

docker start rgb-d-fusion

Hierarki direktori akan terlihat sebagai berikut

 |- tf
   |- manual_datasets
      |-  
         |- test
            |- DEPTH_RENDER_EXR
            |- MASK
            |- PARAM
            |- RENDER
         |- train                     # same hierachy as in test
      |-                    # same hierachy as inv_humas_rendered
   |- GitHub
      |- ConditionalDepthDiffusion    # This Repo
   |- output_runs                     # Auto generated directory to store results
      |- DepthDiffusion
         |- checkpoints               # stores saved model checkpoints
         |- illustrations             # illustrations that are beeing generated during or after training
         |- diffusion_output          # used for inference to store data sampled from the model
      |- SuperResolution              # same hierachy as in DepthDiffusion

Hierarki mungkin dibuat di satu tempat atau di direktori berbeda. Saat memulai wadah buruh pelabuhan, direktori yang berbeda dapat dipasang bersama.

Jalankan skrip Pelatihan, Evaluasi dan/atau Inferensi

Skrip terletak di bawah skrip. Saat ini ada dua jenis model:

Model Difusi Kedalaman , model difusi yang menghasilkan peta kedalaman yang dikondisikan pada gambar RGB
Model Difusi Superresolusi , model difusi yang menghasilkan RGB-D resolusi tinggi dari RGB-D resolusi rendah.

Setiap model memiliki skrip pelatihan, evaluasi, dan inferensi khusus yang ditulis dengan python. Anda dapat memeriksa fungsionalitas dan parameter melalui python

fusi RGB d