【python 人像抠图】利用阿里云api进行对本地图片进行批量人像抠图

阿里云智能视觉开发平台分割抠图
利用阿里云的人像抠图算法，对本地图片进行批量人像抠图，然后批量下载到本地。

# -*- coding: utf-8 -*-
import sys
import requests
import os

from typing import List

from alibabacloud_imageseg20191230.client import Client as imageseg20191230Client
from alibabacloud_tea_openapi import models as open_api_models
from alibabacloud_imageseg20191230 import models as imageseg_20191230_models
from alibabacloud_tea_util import models as util_models
from alibabacloud_tea_util.client import Client as UtilClient


def download(url, filename):
    req = requests.get(url)
    if req.status_code != 200:
        print('DOWNLOAD_ERROR')
        return
    try:
        with open(filename, 'wb') as f:
            # req.content为获取html的内容
            f.write(req.content)
            print('DOWNLOAD_SUCCESS')
    except Exception as e:
        print(e)


class Sample:
    def __init__(self):
        pass

    @staticmethod
    def create_client(
            access_key_id: str,
            access_key_secret: str,
    ) -> imageseg20191230Client:
        """
        使用AK&SK初始化账号Client
        @param access_key_id:
        @param access_key_secret:
        @return: Client
        @throws Exception
        """
        config = open_api_models.Config(
            # 您的 AccessKey ID,
            access_key_id=access_key_id,
            # 您的 AccessKey Secret,
            access_key_secret=access_key_secret
        )
        # 访问的域名
        config.endpoint = f'imageseg.cn-shanghai.aliyuncs.com'
        config.region_id = 'cn-shanghai'
        return imageseg20191230Client(config)

    @staticmethod
    def main(
            dir
    ) -> None:
        files = os.listdir(dir)
        for name in files:
            portion = os.path.splitext(name)
            newname = portion[0] + ".png"
            fullname = os.path.join(dir, name)
            with open(fullname, 'rb') as f:
                # 此处记录 AccessKey ID  AccessKey Secret
                client = Sample.create_client('access_key_id', 'access_key_secret')
                segment_hdbody_request = imageseg_20191230_models.SegmentHDBodyAdvanceRequest()
                segment_hdbody_request.image_urlobject = f
                runtime = util_models.RuntimeOptions()
                try:
                    # 复制代码运行请自行打印 API 的返回值
                    result = client.segment_hdbody_advance(segment_hdbody_request, runtime)
                    result_url = result.body.data.image_url
                    print(result_url)
                    download(url=result_url, filename=newname)
                except Exception as error:
                    # 如有需要，请打印 error
                    error = UtilClient.assert_as_string(error.message)
                    print(error)


if __name__ == '__main__':
    # 待处理图片存储路径
    file_dir = "./dir"
    Sample.main(file_dir)```