mirror of
https://github.com/modelscope/modelscope.git
synced 2025-12-18 17:27:43 +01:00
39 lines
1.4 KiB
Python
39 lines
1.4 KiB
Python
|
# Copyright (c) Alibaba, Inc. and its affiliates.
|
||
|
|
import os.path as osp
|
||
|
|
import shutil
|
||
|
|
import unittest
|
||
|
|
|
||
|
|
from modelscope.fileio import File
|
||
|
|
from modelscope.msdatasets import MsDataset
|
||
|
|
from modelscope.pipelines import pipeline
|
||
|
|
from modelscope.utils.constant import ModelFile, Tasks
|
||
|
|
from modelscope.utils.test_utils import test_level
|
||
|
|
|
||
|
|
|
||
|
|
class Image2ImageTranslationTest(unittest.TestCase):
|
||
|
|
|
||
|
|
@unittest.skipUnless(test_level() >= 0, 'skip test in current test level')
|
||
|
|
def test_run_modelhub(self):
|
||
|
|
r"""We provide three translation modes, i.e., uncropping, colorization and combination.
|
||
|
|
You can pass the following parameters for different mode.
|
||
|
|
1. Uncropping Mode:
|
||
|
|
result = img2img_gen_pipeline(('data/test/images/img2img_input.jpg', 'left', 0, 'result.jpg'))
|
||
|
|
2. Colorization Mode:
|
||
|
|
result = img2img_gen_pipeline(('data/test/images/img2img_input.jpg', 1, 'result.jpg'))
|
||
|
|
3. Combination Mode:
|
||
|
|
just like the following code.
|
||
|
|
"""
|
||
|
|
img2img_gen_pipeline = pipeline(
|
||
|
|
Tasks.image_generation,
|
||
|
|
model='damo/cv_latent_diffusion_image2image_translation')
|
||
|
|
result = img2img_gen_pipeline(
|
||
|
|
('data/test/images/img2img_input_mask.png',
|
||
|
|
'data/test/images/img2img_input_masked_img.png', 2,
|
||
|
|
'result.jpg')) # combination mode
|
||
|
|
|
||
|
|
print(f'output: {result}.')
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
unittest.main()
|