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

myosotis_researches/CcGAN/utils/IMGs_dataset.py

@@ -0,0 +1,33 @@
+import torch
+
+class IMGs_dataset(torch.utils.data.Dataset):
+    def __init__(self, images, labels=None, normalize=False):
+        super(IMGs_dataset, self).__init__()
+
+        self.images = images
+        self.n_images = len(self.images)
+        self.labels = labels
+        if labels is not None:
+            if len(self.images) != len(self.labels):
+                raise Exception('images (' +  str(len(self.images)) +') and labels ('+str(len(self.labels))+') do not have the same length!!!')
+        self.normalize = normalize
+
+
+    def __getitem__(self, index):
+
+        image = self.images[index]
+
+        if self.normalize:
+            image = image/255.0
+            image = (image-0.5)/0.5
+
+        if self.labels is not None:
+            label = self.labels[index]
+            return (image, label)
+        else:
+            return image
+
+    def __len__(self):
+        return self.n_images
+
+__all__ = ["IMGs_dataset"]

myosotis_researches/CcGAN/utils/SimpleProgressBar.py

@@ -0,0 +1,18 @@
+import sys
+
+class SimpleProgressBar():
+    def __init__(self, width=50):
+        self.last_x = -1
+        self.width = width
+
+    def update(self, x):
+        assert 0 <= x <= 100 # `x`: progress in percent ( between 0 and 100)
+        if self.last_x == int(x): return
+        self.last_x = int(x)
+        pointer = int(self.width * (x / 100.0))
+        sys.stdout.write( '\r%d%% [%s]' % (int(x), '#' * pointer + '.' * (self.width - pointer)))
+        sys.stdout.flush()
+        if x == 100:
+            print('')
+
+__all__ = ["SimpleProgressBar"]

myosotis_researches/CcGAN/utils/__init__.py

@@ -1,5 +1,17 @@
 from .print_hdf5 import print_hdf5
 from .concat_image import concat_image
 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
 
-__all__ = ["print_hdf5", "concat_image", "make_h5"]
+__all__ = [
+    "print_hdf5",
+    "concat_image",
+    "make_h5",
+    "SimpleProgressBar",
+    "IMGs_dataset",
+    "PlotLoss",
+    "compute_entropy",
+    "predict_class_labels",
+]

myosotis_researches/CcGAN/utils/train.py

@@ -0,0 +1,65 @@
+import numpy as np
+import torch
+import matplotlib.pyplot as plt
+import matplotlib as mpl
+
+
+def PlotLoss(loss, filename):
+    x_axis = np.arange(start=1, stop=len(loss) + 1)
+    plt.switch_backend("agg")
+    mpl.style.use("seaborn")
+    fig = plt.figure()
+    ax = plt.subplot(111)
+    ax.plot(x_axis, np.array(loss))
+    plt.xlabel("epoch")
+    plt.ylabel("training loss")
+    plt.legend()
+    # ax.legend(loc='upper center', bbox_to_anchor=(0.5, 1.15),  shadow=True, ncol=3)
+    # plt.title('Training Loss')
+    plt.savefig(filename)
+
+
+# compute entropy of class labels; labels is a numpy array
+def compute_entropy(labels, base=None):
+    value, counts = np.unique(labels, return_counts=True)
+    norm_counts = counts / counts.sum()
+    base = np.e if base is None else base
+    return -(norm_counts * np.log(norm_counts) / np.log(base)).sum()
+
+
+def predict_class_labels(net, images, batch_size=500, verbose=False, num_workers=0):
+    net = net.cuda()
+    net.eval()
+
+    n = len(images)
+    if batch_size > n:
+        batch_size = n
+    dataset_pred = IMGs_dataset(images, normalize=False)
+    dataloader_pred = torch.utils.data.DataLoader(
+        dataset_pred, batch_size=batch_size, shuffle=False, num_workers=num_workers
+    )
+
+    class_labels_pred = np.zeros(n + batch_size)
+    with torch.no_grad():
+        nimgs_got = 0
+        if verbose:
+            pb = SimpleProgressBar()
+        for batch_idx, batch_images in enumerate(dataloader_pred):
+            batch_images = batch_images.type(torch.float).cuda()
+            batch_size_curr = len(batch_images)
+
+            outputs, _ = net(batch_images)
+            _, batch_class_labels_pred = torch.max(outputs.data, 1)
+            class_labels_pred[nimgs_got : (nimgs_got + batch_size_curr)] = (
+                batch_class_labels_pred.detach().cpu().numpy().reshape(-1)
+            )
+
+            nimgs_got += batch_size_curr
+            if verbose:
+                pb.update((float(nimgs_got) / n) * 100)
+        # end for batch_idx
+    class_labels_pred = class_labels_pred[0:n]
+    return class_labels_pred
+
+
+__all__ = ["PlotLoss", "compute_entropy", "predict_class_labels"]

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: Myosotis-Researches
-Version: 0.1.4
+Version: 0.1.5
 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.4.dist-info → myosotis_researches-0.1.5.dist-info}/RECORD

@@ -44,12 +44,15 @@ myosotis_researches/CcGAN/train_128_output_10/train_cgan.py,sha256=bYJbBskTpESfC
 myosotis_researches/CcGAN/train_128_output_10/train_cgan_concat.py,sha256=PYctY3IZiHGh4TshXx3mUZBf9su_8NuV_D8InkxKQZ4,8940
 myosotis_researches/CcGAN/train_128_output_10/train_net_for_label_embed.py,sha256=4j6r4_o4rXgAN4MdUQL-TXqZJpbhH7d9gWQR8YzBlXw,6976
 myosotis_researches/CcGAN/train_128_output_10/utils.py,sha256=B-V6ct4WDisVVCOLO0W7VIBL8StPVNJJTZZ2b2NkMFU,3766
-myosotis_researches/CcGAN/utils/__init__.py,sha256=azZ2ZSSmWREoptI_5oQ180HojMoCqv2oleveRswq40w,155
+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/concat_image.py,sha256=BIGKz52Inn9S7M5fBFKye2V9bLJ0DqEQILoOVWAXUiE,2165
 myosotis_researches/CcGAN/utils/make_h5.py,sha256=VtFYjr_i-JktsEW_BvofpilcDmChRmyLykv0VvlMuY0,963
 myosotis_researches/CcGAN/utils/print_hdf5.py,sha256=VvmNAWtMDmg6D9V6ZbSUXrQTKRh9WIJeC4BR_ORJkco,300
-myosotis_researches-0.1.4.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-myosotis_researches-0.1.4.dist-info/METADATA,sha256=O8xsjrUOntGmWZQLl5f7_glRnoCn01smEfF9-B7GA8g,2663
-myosotis_researches-0.1.4.dist-info/WHEEL,sha256=ck4Vq1_RXyvS4Jt6SI0Vz6fyVs4GWg7AINwpsaGEgPE,91
-myosotis_researches-0.1.4.dist-info/top_level.txt,sha256=zxAiMn5eyZNJM28MewTAkgi_RZJMbfWbzVR-KF0LdZE,20
-myosotis_researches-0.1.4.dist-info/RECORD,,
+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,,