Myosotis-Researches 0.1.5__py3-none-any.whl → 0.1.7__py3-none-any.whl

myosotis_researches/CcGAN/models_128/__init__.py

@@ -4,12 +4,4 @@ from .cGAN_concat_SAGAN import cGAN_concat_SAGAN_Generator, cGAN_concat_SAGAN_Di
 from .CcGAN_SAGAN import CcGAN_SAGAN_Generator, CcGAN_SAGAN_Discriminator
 from .ResNet_embed import ResNet18_embed, ResNet34_embed, ResNet50_embed, model_y2h
 from .ResNet_regre_eval import ResNet34_regre_eval
-from .ResNet_class_eval import ResNet34_class_eval
-
-__all__ = [
-    "cGAN_SAGAN_Generator", "cGAN_SAGAN_Discriminator",
-    "cGAN_concat_SAGAN_Generator", "cGAN_concat_SAGAN_Discriminator",
-    "CcGAN_SAGAN_Generator", "CcGAN_SAGAN_Discriminator",
-    "ResNet18_embed", "ResNet34_embed", "ResNet50_embed", "model_y2h",
-    "ResNet34_regre_eval", "ResNet34_class_eval"
-]
+from .ResNet_class_eval import ResNet34_class_eval

myosotis_researches/CcGAN/models_256/__init__.py

@@ -4,12 +4,4 @@ from .cGAN_concat_SAGAN import cGAN_concat_SAGAN_Generator, cGAN_concat_SAGAN_Di
 from .CcGAN_SAGAN import CcGAN_SAGAN_Generator, CcGAN_SAGAN_Discriminator
 from .ResNet_embed import ResNet18_embed, ResNet34_embed, ResNet50_embed, model_y2h
 from .ResNet_regre_eval import ResNet34_regre_eval
-from .ResNet_class_eval import ResNet34_class_eval
-
-__all__ = [
-    "cGAN_SAGAN_Generator", "cGAN_SAGAN_Discriminator",
-    "cGAN_concat_SAGAN_Generator", "cGAN_concat_SAGAN_Discriminator",
-    "CcGAN_SAGAN_Generator", "CcGAN_SAGAN_Discriminator",
-    "ResNet18_embed", "ResNet34_embed", "ResNet50_embed", "model_y2h",
-    "ResNet34_regre_eval", "ResNet34_class_eval"
-]
+from .ResNet_class_eval import ResNet34_class_eval

myosotis_researches/CcGAN/utils/__init__.py

@@ -4,6 +4,7 @@ from .make_h5 import make_h5
 from .SimpleProgressBar import SimpleProgressBar
 from .IMGs_dataset import IMGs_dataset
 from .train import PlotLoss, compute_entropy, predict_class_labels
+from .opts import parse_opts
 
 __all__ = [
     "print_hdf5",
@@ -14,4 +15,5 @@ __all__ = [
     "PlotLoss",
     "compute_entropy",
     "predict_class_labels",
+    "parse_opts"
 ]

myosotis_researches/CcGAN/utils/opts.py

@@ -0,0 +1,89 @@
+import argparse
+
+def parse_opts():
+    parser = argparse.ArgumentParser()
+
+    ''' Overall Settings '''
+    parser.add_argument('--data_path', type=str, default='')
+    parser.add_argument('--output_dir', type=str, default='')
+    parser.add_argument('--eval_ckpt_path', type=str, default='')
+    parser.add_argument('--seed', type=int, default=2021, metavar='S', help='random seed (default: 2020)')
+    parser.add_argument('--num_workers', type=int, default=0)
+
+
+    ''' Dataset '''
+    ## Data split: Ra is split into a train set (the last decimal of the degree is odd) and a test set (the last decimal of the degree is even); the unique labels in two sets do not overlap.
+    parser.add_argument('--data_split', type=str, default='train',
+                        choices=['all', 'train'])
+    parser.add_argument('--min_label', type=float, default=0.0)
+    parser.add_argument('--max_label', type=float, default=90.0)
+    parser.add_argument('--num_channels', type=int, default=3, metavar='N')
+    parser.add_argument('--img_size', type=int, default=128, metavar='N')
+    parser.add_argument('--max_num_img_per_label', type=int, default=50, metavar='N')
+    parser.add_argument('--max_num_img_per_label_after_replica', type=int, default=0, metavar='N')
+    parser.add_argument('--show_real_imgs', action='store_true', default=False)
+    parser.add_argument('--visualize_fake_images', action='store_true', default=False)
+
+
+    ''' GAN settings '''
+    parser.add_argument('--GAN', type=str, default='CcGAN', choices=['cGAN', 'cGAN-concat', 'CcGAN'])
+    parser.add_argument('--GAN_arch', type=str, default='SAGAN', choices=['SAGAN'])
+
+    # label embedding setting
+    parser.add_argument('--net_embed', type=str, default='ResNet34_embed') #ResNetXX_emebed
+    parser.add_argument('--epoch_cnn_embed', type=int, default=200) #epoch of cnn training for label embedding
+    parser.add_argument('--resumeepoch_cnn_embed', type=int, default=0) #epoch of cnn training for label embedding
+    parser.add_argument('--epoch_net_y2h', type=int, default=500)
+    parser.add_argument('--dim_embed', type=int, default=128) #dimension of the embedding space
+    parser.add_argument('--batch_size_embed', type=int, default=256, metavar='N')
+
+    parser.add_argument('--loss_type_gan', type=str, default='hinge')
+    parser.add_argument('--niters_gan', type=int, default=10000, help='number of iterations')
+    parser.add_argument('--resume_niters_gan', type=int, default=0)
+    parser.add_argument('--save_niters_freq', type=int, default=2000, help='frequency of saving checkpoints')
+    parser.add_argument('--lr_g_gan', type=float, default=1e-4, help='learning rate for generator')
+    parser.add_argument('--lr_d_gan', type=float, default=1e-4, help='learning rate for discriminator')
+    parser.add_argument('--dim_gan', type=int, default=128, help='Latent dimension of GAN')
+    parser.add_argument('--batch_size_disc', type=int, default=64)
+    parser.add_argument('--batch_size_gene', type=int, default=64)
+    parser.add_argument('--num_D_steps', type=int, default=4, help='number of Ds updates in one iteration')
+    parser.add_argument('--cGAN_num_classes', type=int, default=20, metavar='N') #bin label into cGAN_num_classes
+    parser.add_argument('--visualize_freq', type=int, default=2000, help='frequency of visualization')
+
+    parser.add_argument('--kernel_sigma', type=float, default=-1.0,
+                        help='If kernel_sigma<0, then use rule-of-thumb formula to compute the sigma.')
+    parser.add_argument('--threshold_type', type=str, default='hard', choices=['soft', 'hard'])
+    parser.add_argument('--kappa', type=float, default=-1)
+    parser.add_argument('--nonzero_soft_weight_threshold', type=float, default=1e-3,
+                        help='threshold for determining nonzero weights for SVDL; we neglect images with too small weights')
+
+    # DiffAugment setting
+    parser.add_argument('--gan_DiffAugment', action='store_true', default=False)
+    parser.add_argument('--gan_DiffAugment_policy', type=str, default='color,translation,cutout')
+
+
+    # evaluation setting
+    '''
+    Four evaluation modes:
+    Mode 1: eval on unique labels used for GAN training;
+    Mode 2. eval on all unique labels in the dataset and when computing FID use all real images in the dataset;
+    Mode 3. eval on all unique labels in the dataset and when computing FID only use real images for GAN training in the dataset (to test SFID's effectiveness on unseen labels);
+    Mode 4. eval on a interval [min_label, max_label] with num_eval_labels labels.
+    '''
+    parser.add_argument('--eval_mode', type=int, default=2)
+    parser.add_argument('--num_eval_labels', type=int, default=-1)
+    parser.add_argument('--samp_batch_size', type=int, default=200)
+    parser.add_argument('--nfake_per_label', type=int, default=200)
+    parser.add_argument('--nreal_per_label', type=int, default=-1)
+    parser.add_argument('--comp_FID', action='store_true', default=False)
+    parser.add_argument('--epoch_FID_CNN', type=int, default=200)
+    parser.add_argument('--FID_radius', type=float, default=0)
+    parser.add_argument('--FID_num_centers', type=int, default=-1)
+    parser.add_argument('--dump_fake_for_NIQE', action='store_true', default=False,
+                        help='Dump fake images for computing NIQE')
+
+    args = parser.parse_args()
+
+    return args
+
+__all__ = ["parse_opts"]

{myosotis_researches-0.1.5.dist-info → myosotis_researches-0.1.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: Myosotis-Researches
-Version: 0.1.5
+Version: 0.1.7
 Summary: A repository for storing my progress of researches.
 Home-page: https://github.com/Zeyu-Xie/Myosotis-Researches
 Author: Zeyu Xie

{myosotis_researches-0.1.5.dist-info → myosotis_researches-0.1.7.dist-info}/RECORD

@@ -8,7 +8,7 @@ myosotis_researches/CcGAN/models_128/CcGAN_SAGAN.py,sha256=uYDngtHoB7frPg2Vs7YCF
 myosotis_researches/CcGAN/models_128/ResNet_class_eval.py,sha256=wa5CPkYzrS0X6kZ6pGHM-GxcGNkSpBdTTqgy5dKVKkU,5131
 myosotis_researches/CcGAN/models_128/ResNet_embed.py,sha256=HKSY-5WWa9jGniOgRoR1WOTfWhR1Dcj6cq2sgznZEbE,6344
 myosotis_researches/CcGAN/models_128/ResNet_regre_eval.py,sha256=VJYJiiwrjf9DvfZrlwOMJJAPu3PlwgFgIddDaRlGsac,6190
-myosotis_researches/CcGAN/models_128/__init__.py,sha256=2knKAbmau6oUkzBPQgyOhhsOzFchnMcXlLP3VS3L07A,768
+myosotis_researches/CcGAN/models_128/__init__.py,sha256=PJQP7ozE9vY23k01he5qvEuGndPZKqxiWWxvgbLDhqg,449
 myosotis_researches/CcGAN/models_128/autoencoder.py,sha256=ugOwBNoSNP4-WiATVkhC4-igRjj6yEY91qU0egpX744,3827
 myosotis_researches/CcGAN/models_128/cGAN_SAGAN.py,sha256=JDr0Ss5osf9m-u34bVN_PvMsvMXkmi2jwPOAnls6EOA,11240
 myosotis_researches/CcGAN/models_128/cGAN_concat_SAGAN.py,sha256=GHAmrNjORXKu-8UqAdP-A5WG-_3BdQUmWsrWD1NX5-w,9634
@@ -16,7 +16,7 @@ myosotis_researches/CcGAN/models_256/CcGAN_SAGAN.py,sha256=ju1dBYhqxl722_eeUGc2m
 myosotis_researches/CcGAN/models_256/ResNet_class_eval.py,sha256=tS5YxIpiFS9tDCNe2IDv1hTZNn40_JBD_nn97MfQJNI,5178
 myosotis_researches/CcGAN/models_256/ResNet_embed.py,sha256=9OcMQ-8nuWEbEbWc9tGaWQtfV1hdnkl0PrTphoGX77c,6295
 myosotis_researches/CcGAN/models_256/ResNet_regre_eval.py,sha256=tHAbRNM9XodyfPsu00ac5KMjcgRH8qdx8AtCN9QGXKc,6269
-myosotis_researches/CcGAN/models_256/__init__.py,sha256=2knKAbmau6oUkzBPQgyOhhsOzFchnMcXlLP3VS3L07A,768
+myosotis_researches/CcGAN/models_256/__init__.py,sha256=PJQP7ozE9vY23k01he5qvEuGndPZKqxiWWxvgbLDhqg,449
 myosotis_researches/CcGAN/models_256/autoencoder.py,sha256=Nv3eSWJVrWaOufoVGe04sZ_KiXFLtu3Y0asZcAdyyj0,4382
 myosotis_researches/CcGAN/models_256/cGAN_SAGAN.py,sha256=wTHVkUcAp07n3lgweKFo6cqd91E_rEqgJrBDbBe6qrg,11510
 myosotis_researches/CcGAN/models_256/cGAN_concat_SAGAN.py,sha256=ZmGEpprDDlFR3dG32LT3NH5yiA1WR8Hg26rcbz42aCQ,9807
@@ -46,13 +46,14 @@ myosotis_researches/CcGAN/train_128_output_10/train_net_for_label_embed.py,sha25
 myosotis_researches/CcGAN/train_128_output_10/utils.py,sha256=B-V6ct4WDisVVCOLO0W7VIBL8StPVNJJTZZ2b2NkMFU,3766
 myosotis_researches/CcGAN/utils/IMGs_dataset.py,sha256=i45PBNSCeAEB5uUG0SluYRTuHWZwH_5ldz2wm6afkYs,927
 myosotis_researches/CcGAN/utils/SimpleProgressBar.py,sha256=S4eD_m6ysHRMHAmRtkTXVRNfXTR8kuHv-d3lUN0BVn4,546
-myosotis_researches/CcGAN/utils/__init__.py,sha256=shSmo-zunolt8zSZ-Cjgv__N2kyflBfrR8UfxvKJqGg,438
+myosotis_researches/CcGAN/utils/__init__.py,sha256=6eJdO4qgHefW606C_ATXg8xhjixeTQHkOdNxBOKACwQ,484
 myosotis_researches/CcGAN/utils/concat_image.py,sha256=BIGKz52Inn9S7M5fBFKye2V9bLJ0DqEQILoOVWAXUiE,2165
 myosotis_researches/CcGAN/utils/make_h5.py,sha256=VtFYjr_i-JktsEW_BvofpilcDmChRmyLykv0VvlMuY0,963
+myosotis_researches/CcGAN/utils/opts.py,sha256=pd7-wknNPBO5hWRpO3YAPmmAsPKgZUUpKc4gWMs6Wto,5397
 myosotis_researches/CcGAN/utils/print_hdf5.py,sha256=VvmNAWtMDmg6D9V6ZbSUXrQTKRh9WIJeC4BR_ORJkco,300
 myosotis_researches/CcGAN/utils/train.py,sha256=NhUee86SkFT7Cq5RG8Fhy0f6WbZNJ5jmomDlhq9FY5I,2140
-myosotis_researches-0.1.5.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-myosotis_researches-0.1.5.dist-info/METADATA,sha256=G8_S3LwrCNALwaZEvMoCWFIIwJRRz1mOoJcmEkN9JUc,2663
-myosotis_researches-0.1.5.dist-info/WHEEL,sha256=ck4Vq1_RXyvS4Jt6SI0Vz6fyVs4GWg7AINwpsaGEgPE,91
-myosotis_researches-0.1.5.dist-info/top_level.txt,sha256=zxAiMn5eyZNJM28MewTAkgi_RZJMbfWbzVR-KF0LdZE,20
-myosotis_researches-0.1.5.dist-info/RECORD,,
+myosotis_researches-0.1.7.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+myosotis_researches-0.1.7.dist-info/METADATA,sha256=Gde6bmI1QC4CsNsEWxgMZ1Eip-dETkF20Z4y1BZTqTw,2663
+myosotis_researches-0.1.7.dist-info/WHEEL,sha256=ck4Vq1_RXyvS4Jt6SI0Vz6fyVs4GWg7AINwpsaGEgPE,91
+myosotis_researches-0.1.7.dist-info/top_level.txt,sha256=zxAiMn5eyZNJM28MewTAkgi_RZJMbfWbzVR-KF0LdZE,20
+myosotis_researches-0.1.7.dist-info/RECORD,,