Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2026-02-24 04:01:10 +01:00
Code Issues Packages Projects Releases Wiki Activity

fix virtural tryon ut error

Browse Source
This commit is contained in:
wenmeng.zwm
2023-08-15 16:47:52 +08:00
parent ee8afd2d62
commit 8db088d2cb
2 changed files with 13 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
modelscope/pipeline_inputs.py
View File

@@ -204,7 +204,14 @@ TASK_INPUTS = {
InputType.IMAGE,
Tasks.video_embedding:
InputType.VIDEO,
Tasks.virtual_try_on: (InputType.IMAGE, InputType.IMAGE, InputType.IMAGE),
Tasks.virtual_try_on: [
(InputType.IMAGE, InputType.IMAGE, InputType.IMAGE),
{
'masked_model': InputType.IMAGE,
'pose': InputType.IMAGE,
'cloth': InputType.IMAGE,
}
],
Tasks.text_driven_segmentation: {
InputKeys.IMAGE: InputType.IMAGE,
InputKeys.TEXT: InputType.TEXT

6
tests/pipelines/test_virtual_try_on.py
View File

@@ -20,7 +20,11 @@ class VirtualTryonTest(unittest.TestCase):
masked_model = Image.open('data/test/images/virtual_tryon_model.jpg')
pose = Image.open('data/test/images/virtual_tryon_pose.jpg')
cloth = Image.open('data/test/images/virtual_tryon_cloth.jpg')
input_imgs = (masked_model, pose, cloth)
input_imgs = {
'masked_model': masked_model,
'pose': pose,
'cloth': cloth,
}
@unittest.skipUnless(test_level() >= 0, 'skip test in current test level')
def test_run_with_model_name(self):