How to render depth images for the albedo diffusion model?

[iris112358]

Hi, Thanks for open-sourcing this awesome project. I can produce reasonable albedo images using loads/chair_depth.png. However, if I use my own rendered depth, the albedo diffusion model performs poorly (one example shown below). May I ask is there any requirement of the depth images used by the albedo diffusion model? Thanks!