pull/1/head
韩天峰 3 years ago
parent 2f32bc11a9
commit ded5ff6966
  1. 12
      cases/ai1.py
  2. 2
      conv.php

@ -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')

@ -119,6 +119,8 @@ class Translator
return 'PyCore::str("' . $attr->s . '")';
case 'Call':
return $this->parseCall($attr);
case 'Subscript':
return $this->parseSubscript($attr);
default:
var_dump(__METHOD__, __LINE__);
debug($attr);

Loading…
Cancel
Save