You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

119 lines
2.8 KiB
Markdown

11 months ago
## Imgae matting
11 months ago
Here are a few effects(omitting mask images)
11 months ago
11 months ago
![image-1](https://github.com/ihmily/image-matting/blob/main/assets/image-1.png)
11 months ago
11 months ago
![image-2](https://github.com/ihmily/image-matting/blob/main/assets/image-2.png)
11 months ago
 
## How to Run
10 months ago
**Method 1: Run from Source Code**
Firstly, you need to download the project code and install the required dependencies.
11 months ago
```
10 months ago
# Python 3.10
11 months ago
11 months ago
git clone https://github.com/ihmily/image-matting.git
cd image-matting
pip install -r requirements.txt
```
10 months ago
Next, use the following command to run the web interface.
11 months ago
```
python app.py
```
10 months ago
Finally, visit http://127.0.0.1:8000/.
 
**Method 2: Run with Docker**
Simply run the following commands after entering the project folder.
Pull the Docker image.
```
10 months ago
docker pull ihmily/image-matting:latest
10 months ago
```
After the image is pulled, run the container.
```
10 months ago
docker run -p 8000:8000 ihmily/image-matting:latest
10 months ago
```
Alternatively, you can build the image yourself.
```
10 months ago
docker build -t image-matting:latest .
docker run -p 8000:8000 image-matting:latest
10 months ago
```
10 months ago
Lastly, visit http://127.0.0.1:8000 to perform online image matting. Feel free to choose the method that suits your preference.
11 months ago
 
11 months ago
## Use API
Please run it before use API
File upload
```
import requests
server = "http://127.0.0.1:8000"
image_path = "image.png"
model_name = "universal" # people,universal
files = {"image": (image_path, open(image_path, "rb"))}
data = {"model": model_name}
response = requests.post(server+'/matting', files=files, data=data)
print(response.text)
json_data = response.json()
image_url = json_data['result_image_url']
mask_url = json_data['mask_image_url']
print("image_url:", server + image_url)
print("mask_url:", server + mask_url)
```
Url upload
```
import requests
server = "http://127.0.0.1:8000"
image_url = "http://your-image-url/demo.png"
data = {"image_url": image_url, "model": "universal"} # people,universal
response = requests.post(server+'/matting/url', json=data)
print(response.text)
json_data = response.json()
image_url = json_data['result_image_url']
mask_url = json_data['mask_image_url']
print("image_url:",server+image_url)
print("mask_url:",server+mask_url)
```
11 months ago
You can freely choose the method you want to upload from above.If you want to get the cropped cutout, you can call `crop_image_by_alpha_channel` function.
 
## Extended Gallery
![image-3](https://github.com/ihmily/image-matting/blob/main/assets/image-3.png)
![image-4](https://github.com/ihmily/image-matting/blob/main/assets/image-4.png)
11 months ago
 
11 months ago
## References
[https://modelscope.cn/models/damo/cv_unet_universal-matting/summary](https://modelscope.cn/models/damo/cv_unet_universal-matting/summary)
11 months ago
[https://modelscope.cn/models/damo/cv_unet_image-matting/summary](https://modelscope.cn/models/damo/cv_unet_image-matting/summary)