renew files

.gitattributes

@@ -54,3 +54,19 @@ pretrained_models/phase3_pretrain_10000.pth filter=lfs diff=lfs merge=lfs -text
 checkpoints/01/epoch_latest.pth filter=lfs diff=lfs merge=lfs -text
 example/source/01.png filter=lfs diff=lfs merge=lfs -text
 example/source/03.png filter=lfs diff=lfs merge=lfs -text
+checkpoints/ filter=lfs diff=lfs merge=lfs -text
+example/ filter=lfs diff=lfs merge=lfs -text
+checkpoints/01 filter=lfs diff=lfs merge=lfs -text
+checkpoints/02 filter=lfs diff=lfs merge=lfs -text
+checkpoints/04 filter=lfs diff=lfs merge=lfs -text
+checkpoints/pop_art filter=lfs diff=lfs merge=lfs -text
+example/outputs filter=lfs diff=lfs merge=lfs -text
+example/reference filter=lfs diff=lfs merge=lfs -text
+checkpoints/03 filter=lfs diff=lfs merge=lfs -text
+checkpoints/watercolor_painting filter=lfs diff=lfs merge=lfs -text
+example/source filter=lfs diff=lfs merge=lfs -text
+checkpoints/04/epoch_latest.pth filter=lfs diff=lfs merge=lfs -text
+checkpoints/pop_art/epoch_latest.pth filter=lfs diff=lfs merge=lfs -text
+checkpoints/watercolor_painting/epoch_latest.pth filter=lfs diff=lfs merge=lfs -text
+checkpoints/02/epoch_latest.pth filter=lfs diff=lfs merge=lfs -text
+checkpoints/03/epoch_latest.pth filter=lfs diff=lfs merge=lfs -text

test_sp2pII.py

@@ -12,12 +12,12 @@ from models.style_based_pix2pixII_model import Stylizer, TrainingPhase
 if __name__ == '__main__':
     # define & parse args
     parser = argparse.ArgumentParser(description='sp2pII test')
-    parser.add_argument('--ckpt', type=str, default='./checkpoints/01/epoch_latest.pth')
+    parser.add_argument('--ckpt', type=str, default='./checkpoints/watercolor_painting/epoch_latest.pth')
     parser.add_argument('--in_folder', type=str, default='./example/source')
-    parser.add_argument('--out_folder', type=str, default='./example/outputs/one-shot')
+    parser.add_argument('--out_folder', type=str, default='./example/outputs/zero-shot/watercolor_painting')
     parser.add_argument('--phase', type=int, default=3)
-    parser.add_argument('--txt_prompt', type=str, default='sketch with black pencil')
-    parser.add_argument('--img_prompt', type=str, default='./example/reference/01.png')
+    parser.add_argument('--txt_prompt', type=str, default='watercolor painting')
+    parser.add_argument('--img_prompt', type=str, default='') # ./example/reference/04.png
     parser.add_argument('--device', type=str, default='cuda:0')
     args = parser.parse_args()
     args.phase = TrainingPhase(args.phase)

train.py

@@ -21,9 +21,9 @@ def main():
     # parse arguments
     parser = argparse.ArgumentParser(description='Style Master')
     parser.add_argument('--cfg_file', type=str, default='./exp/sp2pII-phase4.yaml')
-    parser.add_argument('--image_prompt', type=str, default='./example/reference/01.png')
-    parser.add_argument('--text_prompt', type=str, default='sketch with black pencil')
-    parser.add_argument('--name', type=str, default='01')
+    parser.add_argument('--image_prompt', type=str, default='') #./example/reference/04.png
+    parser.add_argument('--text_prompt', type=str, default='watercolor painting')
+    parser.add_argument('--name', type=str, default='watercolor_painting')
     
     args = parser.parse_args()