CubeInpaint360: Nadir Tripod Removal for 360 Construction Inspection
Method
Cubemap-guided diffusion inpainting: EQR -> cubemap bottom face -> SD Inpainting + LoRA -> re-projection.
Metrics (Best Model: SD15)
- PSNR: 27.07
- SSIM: 0.7794
- LPIPS: 0.0817
Usage
from diffusers import StableDiffusionInpaintPipeline
from peft import PeftModel
pipe = StableDiffusionInpaintPipeline.from_pretrained("runwayml/stable-diffusion-inpainting", torch_dtype=torch.float16, safety_checker=None)
pipe.unet = PeftModel.from_pretrained(pipe.unet, "hiennthp/cubeinpaint360-nadir-removal")
pipe = pipe.to("cuda")
- Downloads last month
- 28
Model tree for hiennthp/cubeinpaint360-nadir-removal
Base model
runwayml/stable-diffusion-inpainting