IORT-Interactive Object Removal Tool based on SAM and LaMa

SAM Mode

Polygon Mode

Install

# create environment
conda create -n iort python=3.9

# activate environment
conda activate iort

pip install torch==2.2 torchvision==0.17.0
pip install -r LaMaProject/requirements.txt

Usage

Download the model checkpoints provided in LaMa (big-lama) and put them into ./checkpoints. For simplicity, you can also go here directly download big-lama and put them into ./checkpoints.

python main.py

To Do

• add visible botton function
• add revise mask function
• output files weidget
• test other segment anything model
• LaMa refinement inpainting mode
• Video object removal
• SAM box mode
• shortcut
• Fix bugs

Acknowledgment

---

• Thanks yatengLG for his awesome project: An Interactive Semi-Automatic Annotation Tool Based on Segment Anything

• Awesome image inpainting method LaMa