parent
2f32bc11a9
commit
ded5ff6966
2 changed files with 14 additions and 0 deletions
@ -0,0 +1,12 @@ |
|||||||
|
from modelscope.pipelines import pipeline |
||||||
|
from modelscope.outputs import OutputKeys |
||||||
|
from PIL import Image |
||||||
|
from background_generation import modelscope_warpper |
||||||
|
|
||||||
|
model = "damo/cv_background_generation_sd" |
||||||
|
pipe = pipeline('background_generation_task', model=model, device='gpu', auto_collate=False, model_revision='v1.1.0') |
||||||
|
main_image = 'https://vision-poster.oss-cn-shanghai.aliyuncs.com/lllcho.lc/data/test_data/demo_example/%E5%8C%96%E5%A6%86%E5%93%81/1c33fc5e8b084269ffdb4e0557c2c3c4.png' |
||||||
|
reference_image = 'https://vision-poster.oss-cn-shanghai.aliyuncs.com/lllcho.lc/data/test_data/5d873b5f64b82bcbb235748347602dce38c6ec1d.jpg' |
||||||
|
out = pipe(main_image, reference_image, num_images_per_prompt=1) |
||||||
|
imgs = out[OutputKeys.OUTPUT_IMGS] |
||||||
|
imgs[0].save(f'result.jpg') |
||||||
Loading…
Reference in new issue