图像局部重绘

from http import HTTPStatus
from urllib.parse import urlparse, unquote
from pathlib import PurePosixPath
import requests
from dashscope import ImageSynthesis

prompt = "a dog wearing red glasses"
model = "wanx-x-painting"
task = "image2image"
extra_input = {
  "base_image_url": "http://synthesis-source.oss-accelerate.aliyuncs.com/lingji/validation/mask2img/demo/source3.jpg",
  "mask_image_url": "http://synthesis-source.oss-accelerate.aliyuncs.com/lingji/validation/mask2img/demo/glasses.png"
}


print('----sync call, please wait a moment----')
rsp = ImageSynthesis.call(model=model,
                          prompt=prompt,
                          n=1,
                          size='1024*1024',
                          task=task,
                          extra_input=extra_input)
if rsp.status_code == HTTPStatus.OK:
  print(rsp.output)
  # save file to current directory
  for result in rsp.output.results:
    file_name = PurePosixPath(unquote(urlparse(result.url).path)).parts[-1]
    with open('./%s' % file_name, 'wb+') as f:
      f.write(requests.get(result.url).content)
else:
  print('sync_call Failed, status_code: %s, code: %s, message: %s' %
        (rsp.status_code, rsp.code, rsp.message))

import com.alibaba.dashscope.aigc.imagesynthesis.ImageSynthesis;
import com.alibaba.dashscope.aigc.imagesynthesis.ImageSynthesisParam;
import com.alibaba.dashscope.aigc.imagesynthesis.ImageSynthesisResult;
import com.alibaba.dashscope.exception.ApiException;
import com.alibaba.dashscope.exception.NoApiKeyException;
import com.alibaba.dashscope.utils.JsonUtils;

import java.util.HashMap;

public class Main {

  public void syncCall() {
    String task = "image2image";
    ImageSynthesis imageSynthesis = new ImageSynthesis(task);
    ImageSynthesisParam param = genImageSynthesis();
    ImageSynthesisResult result = null;
    try {
      System.out.println("---sync call, please wait a moment----");
      result = imageSynthesis.call(param);
    } catch (ApiException | NoApiKeyException e){
      throw new RuntimeException(e.getMessage());
    }
    System.out.println(JsonUtils.toJson(result));
  }

  private ImageSynthesisParam genImageSynthesis(){
    HashMap<String,Object> extraInputMap = new HashMap<>();
    extraInputMap.put("base_image_url", "http://synthesis-source.oss-accelerate.aliyuncs.com/lingji/validation/mask2img/demo/source3.jpg");
    extraInputMap.put("mask_image_url", "http://synthesis-source.oss-accelerate.aliyuncs.com/lingji/validation/mask2img/demo/glasses.png");
    String prompt = "a dog wearing red glasses";
    String model = "wanx-x-painting";
    return ImageSynthesisParam.builder()
            .model(model)
            .prompt(prompt)
            .n(1)
            .size("1024*1024")
            .extraInputs(extraInputMap)
            .build();
  }


  public static void main(String[] args){
    Main text2Image = new Main();
    text2Image.syncCall();
  }

}