Logical bug of `self.img_embed`

[Chieh Hubert Lin]

The persistent attribute self.img_embed is assigned to the current image latent here:

GeoWizard/geowizard/models/geowizard_pipeline.py

Lines 248 to 249 in 0a60193

 if self.img_embed is None: 
     self.__encode_img_embed(input_rgb) 

It is understandable to save AE computation, but the variable is not reset after the depth/normal ensemble. Thus, making a second call to the pipeline inference will result in the wrong input condition.

A simple (but not so clean) fix is to assign self.img_embed=None before return:

GeoWizard/geowizard/models/geowizard_pipeline.py

Lines 199 to 201 in 0a60193

 normal_colored_img = Image.fromarray(normal_colored) 
  
 return DepthNormalPipelineOutput( 

[Xiao Fu]

Thanks, we have fixed it.