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: 25.41
- SSIM: 0.7681
- LPIPS: 0.0983
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-memo")
pipe = pipe.to("cuda")
- Downloads last month
- 11
Model tree for hiennthp/cubeinpaint360-nadir-memo
Base model
runwayml/stable-diffusion-inpainting