feat(erase-object): overhaul object eraser with LaMa inpainting improvements

Update erase-object pipeline, eraser canvas, and inpainting Python script.
Add LaMa model download script and update Dockerfile for model support.
Update multi-file tool routes for consistency.
This commit is contained in:
Siddharth Kumar Sah
2026-04-13 00:48:05 +08:00
parent 92d4d2d9c6
commit 0a506efe24
17 changed files with 405 additions and 95 deletions
+1 -1
View File
@@ -215,7 +215,7 @@ ENV PYTHONWARNINGS=ignore \
# Create non-root user for runtime
RUN groupadd -r stirling && useradd -r -g stirling -d /app -s /sbin/nologin stirling
RUN chown -R stirling:stirling /app /data /tmp/workspace /opt/venv
RUN chown -R stirling:stirling /app /data /tmp/workspace /opt/venv /opt/models
# Entrypoint fixes volume permissions then drops to stirling via gosu
COPY docker/entrypoint.sh /usr/local/bin/entrypoint.sh
+29
View File
@@ -13,6 +13,11 @@ os.environ["PADDLE_DEVICE"] = "cpu"
os.environ["FLAGS_use_cuda"] = "0"
os.environ["CUDA_VISIBLE_DEVICES"] = ""
LAMA_MODEL_DIR = "/opt/models/lama"
LAMA_MODEL_URL = "https://huggingface.co/Carve/LaMa-ONNX/resolve/main/lama_fp32.onnx"
LAMA_MODEL_PATH = os.path.join(LAMA_MODEL_DIR, "lama_fp32.onnx")
LAMA_MIN_SIZE = 100_000_000 # ~200 MB
REALESRGAN_MODEL_DIR = "/opt/models/realesrgan"
REALESRGAN_MODEL_URL = (
"https://github.com/xinntao/Real-ESRGAN/releases/download/v0.1.0/RealESRGAN_x4plus.pth"
@@ -98,6 +103,20 @@ def download_rembg_models():
print(f"All {len(REMBG_MODELS)} rembg models downloaded.\n")
def download_lama_model():
"""Download LaMa ONNX inpainting model from HuggingFace."""
print("=== Downloading LaMa ONNX model ===")
os.makedirs(LAMA_MODEL_DIR, exist_ok=True)
print(f" Downloading from {LAMA_MODEL_URL}...")
urllib.request.urlretrieve(LAMA_MODEL_URL, LAMA_MODEL_PATH)
size = os.path.getsize(LAMA_MODEL_PATH)
assert size > LAMA_MIN_SIZE, (
f"LaMa model too small: {size} bytes (expected > {LAMA_MIN_SIZE})"
)
print(f" lama_fp32.onnx downloaded ({size / 1_000_000:.1f} MB)\n")
def download_realesrgan_model():
"""Download RealESRGAN_x4plus.pth pretrained weights."""
print("=== Downloading RealESRGAN model ===")
@@ -196,6 +215,15 @@ def smoke_test():
import mediapipe as mp
print(" MediaPipe import OK")
# LaMa model file must exist
assert os.path.exists(LAMA_MODEL_PATH), (
f"LaMa model missing: {LAMA_MODEL_PATH}"
)
assert os.path.getsize(LAMA_MODEL_PATH) > LAMA_MIN_SIZE, (
"LaMa model file is too small"
)
print(" LaMa ONNX model file verified")
# RealESRGAN model file must exist
assert os.path.exists(REALESRGAN_MODEL_PATH), (
f"RealESRGAN model missing: {REALESRGAN_MODEL_PATH}"
@@ -232,6 +260,7 @@ def smoke_test():
def main():
print("Pre-downloading all ML models...\n")
download_lama_model()
download_rembg_models()
download_realesrgan_model()
download_gfpgan_model()