https://github.com/taylor-eos/sam

Create segmentation masks for automatically selecting objects in your images
https://github.com/taylor-eos/sam

Last synced: 17 days ago
JSON representation

Create segmentation masks for automatically selecting objects in your images

• Host: GitHub
• URL: https://github.com/taylor-eos/sam
• Owner: Taylor-eOS
• Created: 2025-02-23T20:20:17.000Z (over 1 year ago)
• Default Branch: main
• Last Pushed: 2025-02-23T20:33:52.000Z (over 1 year ago)
• Last Synced: 2025-02-23T21:26:27.492Z (over 1 year ago)
• Language: Python
• Size: 4.88 KB
• Stars: 0
• Watchers: 1
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

README

          
**Python Script to create layer masks in [Segment Anything](https://huggingface.co/docs/transformers/model_doc/sam) (SAM)**

![sam](https://github.com/user-attachments/assets/2e47564c-ead7-4c11-82bc-5382a133f15a)

### **Instructions:**

Install dependencies with `pip install -r requirements.txt`.

Download the model from `https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth` (version name might have changed) and place it in the same folder as the script.

Open your image and find the coordinates for the object you want to select. (Draw a box around it and hover your mouse over the boundaries.)

Run the script with `python sam.py input.jpg x0 y0 x1 y1`, replacing `input.jpg` with your image file and entering the bounding box coordinates as four integers: `x0` and `y0` for the top-left corner, `x1` and `y1` for the bottom-right corner.

These define the rectangular area in which the model will select an object.

The script saves `segmentation_mask.png` in the same folder.

To use your black-and-white segmentation mask in GIMP:

1. **Open the Segmentation Mask Image**  

   - Open the black-and-white segmentation mask (`File > Open as Layers`) so it appears as a separate layer above your original image.

2. **Copy the Mask Image**  

   - Select the segmentation mask layer in the **Layers** panel.

   - Press **Ctrl + A** to select all.

   - Press **Ctrl + C** to copy it.

3. **Add a Layer Mask to the Original Image**  

   - Click on your original image’s layer in the **Layers** panel.

   - Right-click it and choose **Add Layer Mask**.

   - In the dialog, select **White (full opacity)** and click **Add**.

   - If the mask is inverted, click **Colors**, **Invert**.

   - Click anywhere else in the image with the `Rectangle` tool to remove the selection. Delete or hide the black-and-white upper layer.

4. **Paste the Segmentation Mask into the Layer Mask**  

   - Press **Ctrl + V** to paste the segmentation mask into it.

   - In the **Layers** panel, click the **Anchor Layer** button or press **Ctrl + H** to apply the mask.

   - Click anywhere else in the image with the `Rectangle` tool to merge the floating selection.

5. **Use the mask**

   - Go to **Select > By Color**.

   - Click on the part of the mask you want to cut out.

   - Adjust the **Threshold** if necessary (higher means selecting more shades of gray).

   - Click on your original image layer in the **Layers** panel, and copy (**Ctrl + V**) the selection.

   - In **Layers**, click **New Layer** button, create a new **Transparency** layer.

   - Paste in the floating selection. Click anywhere else to remove the selection.

   - Click **Layer**, **Crop content**.