Update README.md

main
Hmily 11 months ago committed by GitHub
parent 8d566e3559
commit 33fcde8d6f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -13,6 +13,8 @@ Here are a few effects
Firstly, you need to download the project code and install the required dependencies
```
# python 3.10
git clone https://github.com/ihmily/image-matting.git
cd image-matting
pip install -r requirements.txt
@ -28,6 +30,50 @@ Finally, you can visit http://127.0.0.1:8000
 
## 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)
```
You can freely choose the method you want to upload from above.
 
## References
[https://modelscope.cn/models/damo/cv_unet_universal-matting/summary](https://modelscope.cn/models/damo/cv_unet_universal-matting/summary)

Loading…
Cancel
Save