diff --git a/asset/images/inpainting_text/ex3_scene_im.jpg b/asset/images/inpainting_text/ex3_scene_im.jpg index ef76b8f..de2ea26 100644 Binary files a/asset/images/inpainting_text/ex3_scene_im.jpg and b/asset/images/inpainting_text/ex3_scene_im.jpg differ diff --git a/asset/images/inpainting_text/ex3_scene_mask.jpg b/asset/images/inpainting_text/ex3_scene_mask.jpg index 70ec5a5..5bed2f6 100644 Binary files a/asset/images/inpainting_text/ex3_scene_mask.jpg and b/asset/images/inpainting_text/ex3_scene_mask.jpg differ diff --git a/asset/images/inpainting_text/ex3_scene_mask2.jpg b/asset/images/inpainting_text/ex3_scene_mask2.jpg index 159bd6f..623b92c 100644 Binary files a/asset/images/inpainting_text/ex3_scene_mask2.jpg and b/asset/images/inpainting_text/ex3_scene_mask2.jpg differ diff --git a/asset/images/inpainting_text/inpainting_text.jpeg b/asset/images/inpainting_text/inpainting_text.jpeg deleted file mode 100644 index 174dba0..0000000 Binary files a/asset/images/inpainting_text/inpainting_text.jpeg and /dev/null differ diff --git a/asset/images/inpainting_text/inpainting_text.jpg b/asset/images/inpainting_text/inpainting_text.jpg new file mode 100644 index 0000000..c4554be Binary files /dev/null and b/asset/images/inpainting_text/inpainting_text.jpg differ diff --git a/asset/images/inpainting_text/inpainting_text2.jpeg b/asset/images/inpainting_text/inpainting_text2.jpeg deleted file mode 100644 index 7f827ad..0000000 Binary files a/asset/images/inpainting_text/inpainting_text2.jpeg and /dev/null differ diff --git a/asset/images/inpainting_text/inpainting_text2.jpg b/asset/images/inpainting_text/inpainting_text2.jpg new file mode 100644 index 0000000..e73b474 Binary files /dev/null and b/asset/images/inpainting_text/inpainting_text2.jpg differ diff --git a/asset/images/inpainting_text_ref/ex4_scene_im.jpg b/asset/images/inpainting_text_ref/ex4_scene_im.jpg index 6233409..0bcdbe1 100644 Binary files a/asset/images/inpainting_text_ref/ex4_scene_im.jpg and b/asset/images/inpainting_text_ref/ex4_scene_im.jpg differ diff --git a/asset/images/inpainting_text_ref/ex4_scene_mask.jpg b/asset/images/inpainting_text_ref/ex4_scene_mask.jpg index 9c1fe96..6f1e831 100644 Binary files a/asset/images/inpainting_text_ref/ex4_scene_mask.jpg and b/asset/images/inpainting_text_ref/ex4_scene_mask.jpg differ diff --git a/asset/images/inpainting_text_ref/ex4_subject_im.jpg b/asset/images/inpainting_text_ref/ex4_subject_im.jpg index 27f0835..69dcfc5 100644 Binary files a/asset/images/inpainting_text_ref/ex4_subject_im.jpg and b/asset/images/inpainting_text_ref/ex4_subject_im.jpg differ diff --git a/asset/images/inpainting_text_ref/ex4_subject_mask.jpg b/asset/images/inpainting_text_ref/ex4_subject_mask.jpg index e3124cf..21ba6dd 100644 Binary files a/asset/images/inpainting_text_ref/ex4_subject_mask.jpg and b/asset/images/inpainting_text_ref/ex4_subject_mask.jpg differ diff --git a/asset/images/inpainting_text_ref/inpainting_text_ref.jpeg b/asset/images/inpainting_text_ref/inpainting_text_ref.jpeg deleted file mode 100644 index 9f61239..0000000 Binary files a/asset/images/inpainting_text_ref/inpainting_text_ref.jpeg and /dev/null differ diff --git a/asset/images/inpainting_text_ref/inpainting_text_ref.jpg b/asset/images/inpainting_text_ref/inpainting_text_ref.jpg new file mode 100644 index 0000000..ccb140a Binary files /dev/null and b/asset/images/inpainting_text_ref/inpainting_text_ref.jpg differ diff --git a/asset/images/largen.gif b/asset/images/largen.gif deleted file mode 100644 index bd07415..0000000 Binary files a/asset/images/largen.gif and /dev/null differ diff --git a/asset/images/virtual_try_on/ex2_scene_mask.jpg b/asset/images/virtual_try_on/ex2_scene_mask.jpg index f800425..c741567 100644 Binary files a/asset/images/virtual_try_on/ex2_scene_mask.jpg and b/asset/images/virtual_try_on/ex2_scene_mask.jpg differ diff --git a/asset/images/virtual_try_on/ex2_subject_mask.jpg b/asset/images/virtual_try_on/ex2_subject_mask.jpg index 58678c0..affcf50 100644 Binary files a/asset/images/virtual_try_on/ex2_subject_mask.jpg and b/asset/images/virtual_try_on/ex2_subject_mask.jpg differ diff --git a/asset/images/virtual_try_on/model.jpg b/asset/images/virtual_try_on/model.jpg index d0b589e..b076637 100644 Binary files a/asset/images/virtual_try_on/model.jpg and b/asset/images/virtual_try_on/model.jpg differ diff --git a/asset/images/virtual_try_on/try_on_out.jpeg b/asset/images/virtual_try_on/try_on_out.jpeg deleted file mode 100644 index 35c81e8..0000000 Binary files a/asset/images/virtual_try_on/try_on_out.jpeg and /dev/null differ diff --git a/asset/images/virtual_try_on/try_on_out.jpg b/asset/images/virtual_try_on/try_on_out.jpg new file mode 100644 index 0000000..c93e0a1 Binary files /dev/null and b/asset/images/virtual_try_on/try_on_out.jpg differ diff --git a/asset/images/virtual_try_on/tshirt.jpg b/asset/images/virtual_try_on/tshirt.jpg index 462db84..36b5eb3 100644 Binary files a/asset/images/virtual_try_on/tshirt.jpg and b/asset/images/virtual_try_on/tshirt.jpg differ diff --git a/asset/images/zoom_out/ex1_scene_im.jpeg b/asset/images/zoom_out/ex1_scene_im.jpeg deleted file mode 100644 index 089e453..0000000 Binary files a/asset/images/zoom_out/ex1_scene_im.jpeg and /dev/null differ diff --git a/asset/images/zoom_out/ex1_scene_im.jpg b/asset/images/zoom_out/ex1_scene_im.jpg new file mode 100644 index 0000000..f53852a Binary files /dev/null and b/asset/images/zoom_out/ex1_scene_im.jpg differ diff --git a/asset/images/zoom_out/ex1_zoom_out1.jpeg b/asset/images/zoom_out/ex1_zoom_out1.jpeg deleted file mode 100644 index ccd5cf4..0000000 Binary files a/asset/images/zoom_out/ex1_zoom_out1.jpeg and /dev/null differ diff --git a/asset/images/zoom_out/ex1_zoom_out1.jpg b/asset/images/zoom_out/ex1_zoom_out1.jpg new file mode 100644 index 0000000..c0ab62d Binary files /dev/null and b/asset/images/zoom_out/ex1_zoom_out1.jpg differ diff --git a/asset/images/zoom_out/ex1_zoom_out2.jpeg b/asset/images/zoom_out/ex1_zoom_out2.jpeg deleted file mode 100644 index 25e1c10..0000000 Binary files a/asset/images/zoom_out/ex1_zoom_out2.jpeg and /dev/null differ diff --git a/asset/images/zoom_out/ex1_zoom_out2.jpg b/asset/images/zoom_out/ex1_zoom_out2.jpg new file mode 100644 index 0000000..3c30e90 Binary files /dev/null and b/asset/images/zoom_out/ex1_zoom_out2.jpg differ diff --git a/asset/images/zoom_out/ex1_zoom_out3.jpeg b/asset/images/zoom_out/ex1_zoom_out3.jpeg deleted file mode 100644 index 662f996..0000000 Binary files a/asset/images/zoom_out/ex1_zoom_out3.jpeg and /dev/null differ diff --git a/asset/images/zoom_out/ex1_zoom_out3.jpg b/asset/images/zoom_out/ex1_zoom_out3.jpg new file mode 100644 index 0000000..ce2949f Binary files /dev/null and b/asset/images/zoom_out/ex1_zoom_out3.jpg differ diff --git a/asset/images/zoom_out/ex1_zoom_out4.jpeg b/asset/images/zoom_out/ex1_zoom_out4.jpeg deleted file mode 100644 index 8911c1d..0000000 Binary files a/asset/images/zoom_out/ex1_zoom_out4.jpeg and /dev/null differ diff --git a/asset/images/zoom_out/ex1_zoom_out4.jpg b/asset/images/zoom_out/ex1_zoom_out4.jpg new file mode 100644 index 0000000..77738e3 Binary files /dev/null and b/asset/images/zoom_out/ex1_zoom_out4.jpg differ diff --git a/example/__init__.py b/example/__init__.py index c2df617..cc26a06 100644 --- a/example/__init__.py +++ b/example/__init__.py @@ -1,2 +1,2 @@ # -*- coding: utf-8 -*- -from . import impls +# Copyright (c) Alibaba, Inc. and its affiliates. diff --git a/example/impls/dataset/__init__.py b/example/classifier/__init__.py similarity index 62% rename from example/impls/dataset/__init__.py rename to example/classifier/__init__.py index 1635bb3..50d63d8 100644 --- a/example/impls/dataset/__init__.py +++ b/example/classifier/__init__.py @@ -1,2 +1,3 @@ # -*- coding: utf-8 -*- +# Copyright (c) Alibaba, Inc. and its affiliates. from .classifier_dataset import ImageClassifyExampleDataset diff --git a/example/example.yaml b/example/classifier/classifier.yaml similarity index 100% rename from example/example.yaml rename to example/classifier/classifier.yaml diff --git a/example/impls/dataset/classifier_dataset.py b/example/classifier/classifier_dataset.py similarity index 100% rename from example/impls/dataset/classifier_dataset.py rename to example/classifier/classifier_dataset.py diff --git a/example/run.py b/example/classifier/run.py similarity index 100% rename from example/run.py rename to example/classifier/run.py diff --git a/example/impls/__init__.py b/example/impls/__init__.py deleted file mode 100644 index 945fb86..0000000 --- a/example/impls/__init__.py +++ /dev/null @@ -1,2 +0,0 @@ -# -*- coding: utf-8 -*- -from . import dataset diff --git a/readme.md b/readme.md index 44afbc2..2467a43 100644 --- a/readme.md +++ b/readme.md @@ -169,10 +169,10 @@ python scepter/tools/run_inference.py --cfg scepter/methods/scedit/ctr/sd21_768_ ``` ### Customize Modules -Refer to `example`, build your modules in `example/impls`. +Refer to `example`, build the modules of your task in `example/{task}`. ```python -cd example -python run.py --cfg example.yaml +cd example/classifier +python run.py --cfg classifier.yaml ``` @@ -199,7 +199,7 @@ Therefore, subsequent startups will become much faster (about one minute) as dow * LAR-Gen: we release `zoom-out`, `virtual try on`, `inpainting(text guided)`, `inpainting(text + reference image guided)` image editing capabilities. Please note that the **Data Preprocess** button must be clicked before clicking the **Generate** button.

- +

### Modelscope Studio @@ -208,7 +208,7 @@ We deploy a work studio on Modelscope that includes only the inference tab, plea ## 🖼️ Gallery -### LAR-Gen: Zoom-Out +### LAR-Gen: Zoom Out @@ -218,15 +218,15 @@ We deploy a work studio on Modelscope that includes only the inference tab, plea - - - - - + + + + +
Origin Image
Prompt: a temple on fire		Zoom-Out
CenterAround:0.75
-### LAR-Gen: Virtual-Try-on +### LAR-Gen: Virtual Try-on @@ -240,11 +240,11 @@ We deploy a work studio on Modelscope that includes only the inference tab, plea - +
Model Image
-### LAR-Gen: Inpainting(Text-guided) +### LAR-Gen: Inpainting (Text guided) @@ -256,13 +256,13 @@ We deploy a work studio on Modelscope that includes only the inference tab, plea - + - +
Origin Image
Prompt: a blue and white porcelain
-### LAR-Gen: Inpainting(Text + Reference Image Guided) +### LAR-Gen: Inpainting (Text and Subject guided) @@ -276,7 +276,7 @@ We deploy a work studio on Modelscope that includes only the inference tab, plea - +
Origin Image
Prompt: a dog wearing sunglasses
diff --git a/scepter/modules/solver/hooks/data_probe.py b/scepter/modules/solver/hooks/data_probe.py index ea98326..b51aaeb 100644 --- a/scepter/modules/solver/hooks/data_probe.py +++ b/scepter/modules/solver/hooks/data_probe.py @@ -47,7 +47,7 @@ class ProbeDataHook(Hook): if we.rank == 0: save_folder = os.path.join( solver.work_dir, - f'{solver.mode}_probe/{self.save_probe_prefix}-{solver.total_iter}' + f'{solver.mode}_probe/{self.save_name_prefix}-{solver.total_iter}' ) ret_data = {} for k, v in probe_dict.items(): diff --git a/scepter/studio/tuner_manager/__init__.py b/scepter/studio/tuner_manager/__init__.py index e69de29..cc26a06 100644 --- a/scepter/studio/tuner_manager/__init__.py +++ b/scepter/studio/tuner_manager/__init__.py @@ -0,0 +1,2 @@ +# -*- coding: utf-8 -*- +# Copyright (c) Alibaba, Inc. and its affiliates. diff --git a/scepter/studio/tuner_manager/manager_ui/__init__.py b/scepter/studio/tuner_manager/manager_ui/__init__.py index e69de29..cc26a06 100644 --- a/scepter/studio/tuner_manager/manager_ui/__init__.py +++ b/scepter/studio/tuner_manager/manager_ui/__init__.py @@ -0,0 +1,2 @@ +# -*- coding: utf-8 -*- +# Copyright (c) Alibaba, Inc. and its affiliates. diff --git a/scepter/studio/tuner_manager/utils/__init__.py b/scepter/studio/tuner_manager/utils/__init__.py index e69de29..cc26a06 100644 --- a/scepter/studio/tuner_manager/utils/__init__.py +++ b/scepter/studio/tuner_manager/utils/__init__.py @@ -0,0 +1,2 @@ +# -*- coding: utf-8 -*- +# Copyright (c) Alibaba, Inc. and its affiliates.