Differences

This shows you the differences between two versions of the page.

Link to this comparison view

Next revision		Previous revision
stylegan_notes [2026-08-03 08:51] – created nikstylegan_notes [2026-08-03 09:01] (current) – nik
Line 1: Line 1:
 ==== StyleGAN Notes ==== ==== StyleGAN Notes ====
  
-styleGAN2-ADA experiments Feb 2021 (and so on and so on https://github.com/NVlabs/stylegan2-ada-pytorch)+styleGAN2-ADA experiments Feb 2021 (and so on and so on...) 
 + 
 +  * using ''foam-repo/F20/web/src/flickr/img'' 
 +    * two data folders (prepared with dataset_tool.py) 
 +      * ''1024-foam-f20'' all the images tagged with f15 machine tags 
 +      * ''1024-zzkt-f20'' all the images tagged with f15 machine tags and zzkt images 
 +  * transfer learning from flickr-1024 network (i.e ''ffhq.pkl'') via ''--resume=ffhq1024'' 
 +  * training strategy 
 +    * start with ''--cfg=auto --aug=ada --metrics=fid50k_full'' 
 +    * compare unmirrored with → ''--mirror=1'' 
 +    * try ''--augpipe=bgcfnc'' 
 +    * try various gamma levels ''--gamma=10'' 
  
  
Line 31: Line 43:
  
 <code shell> <code shell>
-mogrify -define png:color-type=2 /root/data/1024x1024-foam-flickr/*.png+mogrify -define png:color-type=2 data/1024x1024-foam-flickr/*.png
 </code> </code>
  
 <code shell> <code shell>
-python3 dataset_tool.py --source=/root/data/1024x1024 --dest=/root/data/foam-flickr/+python3 dataset_tool.py --source=data/1024x1024 --dest=data/foam-flickr/
 </code> </code>
  
 <code shell> <code shell>
-python3 train.py --outdir=/root/results/flickr4/ \ +python3 train.py --outdir=results/flickr4/ \ 
-                 --data=/root/data/512x512/ \+                 --data=data/512x512/ \
                  --gpus=1 \                  --gpus=1 \
                  --resume=ffhq512 --snap=10                  --resume=ffhq512 --snap=10
Line 50: Line 62:
  
 <code shell> <code shell>
-python3 generate.py --outdir=/root/results/ \+python3 generate.py --outdir=results/ \
                     --trunc=1 \                     --trunc=1 \
-                    --network=/root/results/flickr3/00009--auto1-resumeffhq1024/network-snapshot-000000.pkl \+                    --network=results/flickr3/00009--auto1-resumeffhq1024/network-snapshot-000000.pkl \
                     --seeds=85,265,297,849                     --seeds=85,265,297,849
 </code> </code>
  
 <code shell> <code shell>
-generate.py --outdir=/root/results/flickr5/generated/000041 --network=/root/results/flickr5/00002-512x512-auto1-resumecustom/network-snapshot-000040.pkl --seeds=1100-1200+generate.py --outdir=results/flickr5/generated/000041 \ 
 +--network=results/flickr5/00002-512x512-auto1-resumecustom/network-snapshot-000040.pkl \ 
 +--seeds=1100-1200
 </code> </code>
  
Line 64: Line 78:
  
 <code shell> <code shell>
-python3 style_mixing.py --outdir=/root/results/ \ +python3 style_mixing.py --outdir=results/ \ 
-                        --network=/root/results/flickr3/00009--auto1-resumeffhq1024/network-snapshot-000000.pkl \+                        --network=results/flickr3/00009--auto1-resumeffhq1024/network-snapshot-000000.pkl \
                         --rows=85,100,75,458,1500 \                         --rows=85,100,75,458,1500 \
                         --cols=55,821,1789,293                         --cols=55,821,1789,293
Line 71: Line 85:
  
 <code shell> <code shell>
-python3 style_mixing.py --outdir=/root/results/flickr5/generated/000000 --network=/root/results/flickr5/00001-512x512-auto1-resumecustom/network-snapshot-000000.pkl --rows=600-610 --cols=108-118+style_mixing.py --outdir=results/flickr5/generated/000000 --network=results/flickr5/00001-512x512-auto1-resumecustom/network-snapshot-000000.pkl --rows=600-610 --cols=108-118
 </code> </code>
  
Line 77: Line 91:
  
 <code shell> <code shell>
-python3 projector.py --outdir=/root/results/flickr6/generated/100000 \ +python3 projector.py --outdir=results/flickr6/generated/100000 \ 
-    --target=/root/data/_transfer_test/41018347952_71cbe8d49e_k\(1\).jpg \ +    --target=data/_transfer_test/41018347952_71cbe8d49e_k\(1\).jpg \ 
-    --network=/root/results/flickr6/00003-512x512-auto1-resumecustom/network-snapshot-000200.pkl+    --network=results/flickr6/00003-512x512-auto1-resumecustom/network-snapshot-000200.pkl
 </code> </code>
  
Line 87: Line 101:
  
 <code shell> <code shell>
-python3 train.py --outdir=/root/results/flickr3/ \ +python3 train.py --outdir=results/flickr3/ \ 
-                 --data=/root/data/foam-flickr/ \+                 --data=data/foam-flickr/ \
           --gpus=1 \           --gpus=1 \
       --resume=ffhq1024 \       --resume=ffhq1024 \
Line 95: Line 109:
  
 <code shell> <code shell>
-python3 train.py --outdir=/root/results/flickr3/ --data=/root/data/foam-flickr/ --gpus=1 --resume=ffhq1024 --snap=10+python3 train.py --outdir=results/flickr3/ --data=data/foam-flickr/ --gpus=1 --resume=ffhq1024 --snap=10
  
 Training options: Training options:
Line 108: Line 122:
   "training_set_kwargs": {   "training_set_kwargs": {
     "class_name": "training.dataset.ImageFolderDataset",     "class_name": "training.dataset.ImageFolderDataset",
-    "path": "/root/data/foam-flickr/",+    "path": "data/foam-flickr/",
     "use_labels": false,     "use_labels": false,
     "max_size": 1140,     "max_size": 1140,
Line 190: Line 204:
   "resume_pkl": "https://nvlabs-fi-cdn.nvidia.com/stylegan2-ada-pytorch/pretrained/transfer-learning-source-nets/ffhq-res1024-mirror-stylegan2-noaug.pkl",   "resume_pkl": "https://nvlabs-fi-cdn.nvidia.com/stylegan2-ada-pytorch/pretrained/transfer-learning-source-nets/ffhq-res1024-mirror-stylegan2-noaug.pkl",
   "ada_kimg": 100,   "ada_kimg": 100,
-  "run_dir": "/root/results/flickr3/00009--auto1-resumeffhq1024"+  "run_dir": "results/flickr3/00009--auto1-resumeffhq1024"
 } }
  
-Output directory:   /root/results/flickr3/00009--auto1-resumeffhq1024 +Output directory:   results/flickr3/00009--auto1-resumeffhq1024 
-Training data:      /root/data/foam-flickr/+Training data:      data/foam-flickr/
 Training duration:  25000 kimg Training duration:  25000 kimg
 Number of GPUs:     1 Number of GPUs:     1
Line 376: Line 390:
  
 <code python> <code python>
-python3 train.py --outdir=/root/results/flickr3/ --data=/root/data/foam-flickr/ --gpus=1 --resume=ffhq1024 --snap=10+python3 train.py --outdir=results/flickr3/ --data=data/foam-flickr/ --gpus=1 --resume=ffhq1024 --snap=10
  
 tick 11    kimg 44.0     time 2h 53m 35s   sec/tick 784.5   sec/kimg 196.13  maintenance 897.6  cpumem 3.93   gpumem 7.96   augment 0.209 tick 11    kimg 44.0     time 2h 53m 35s   sec/tick 784.5   sec/kimg 196.13  maintenance 897.6  cpumem 3.93   gpumem 7.96   augment 0.209
Line 424: Line 438:
  
 <code shell> <code shell>
-python3 train.py --outdir=/root/results/foam-flickr/20210316 \ +python3 train.py --outdir=results/foam-flickr/20210316 \ 
-        --data=/root/data/1024x1024-foam-flickr \ +        --data=data/1024x1024-foam-flickr \ 
-        --resume=/root/results/foam-flickr/20210310/00001-1024x1024-foam-flickr-auto2-resumecustom/network-snapshot-005040.pkl \+        --resume=results/foam-flickr/20210310/00001-1024x1024-foam-flickr-auto2-resumecustom/network-snapshot-005040.pkl \
         --snap=20 \         --snap=20 \
         --gpus=2 \         --gpus=2 \
Line 435: Line 449:
  
 <code shell> <code shell>
-python3 train.py --outdir=/root/results/foam-flickr/20210316 \ +python3 train.py --outdir=results/foam-flickr/20210316 \ 
-        --data=/root/data/1024x1024-foam-flickr \ +        --data=data/1024x1024-foam-flickr \ 
-        --resume=/root/results/foam-flickr/20210310/00001-1024x1024-foam-flickr-auto2-resumecustom/network-snapshot-005040.pkl \+        --resume=results/foam-flickr/20210310/00001-1024x1024-foam-flickr-auto2-resumecustom/network-snapshot-005040.pkl \
         --snap=20 \         --snap=20 \
         --gpus=2 \         --gpus=2 \
Line 445: Line 459:
  
 <code shell> <code shell>
-python3 generate.py \ --network=/root/results/foam-flickr/20210316/00000-1024x1024-foam-flickr-auto2-gamma2-batch16-resumecustom/network-snapshot-000000.pkl \ +python3 generate.py \  
- --outdir=/root/results/foam-flickr/20210316/generated/foam-000000 \+--network=results/foam-flickr/20210316/00000-1024x1024-foam-flickr-auto2-gamma2-batch16-resumecustom/network-snapshot-000000.pkl \ 
 + --outdir=results/foam-flickr/20210316/generated/foam-000000 \
  --seeds=1-10000  --seeds=1-10000
 </code> </code>
  
  

User Tools

Page Tools

Site Tools