• Search by Module
• Search by Words
• Search Projects

• Java
• Python
• JavaScript
• TypeScript
• C++
• Scala
• Blog

• Deep-SAD-PyTorch-master
  • src
    • datasets
      • cifar10.py
      • mnist.py
      • fmnist.py
      • preprocessing.py
      • __init__.py
      • main.py
      • odds.py
    • utils
      • misc.py
      • config.py
      • __init__.py
      • visualization
        • plot_images_grid.py
    • optim
      • ae_trainer.py
      • SemiDGM_trainer.py
      • vae_trainer.py
      • DeepSAD_trainer.py
      • variational.py
      • __init__.py
    • baseline_SemiDGM.py
    • baseline_isoforest.py
    • baseline_ocsvm.py
    • base
      • base_net.py
      • base_trainer.py
      • __init__.py
      • odds_dataset.py
      • torchvision_dataset.py
      • base_dataset.py
    • networks
      • cifar10_LeNet.py
      • layers
        • stochastic.py
        • standard.py
      • vae.py
      • mnist_LeNet.py
      • fmnist_LeNet.py
      • __init__.py
      • main.py
      • inference
        • distributions.py
      • mlp.py
      • dgm.py
    • baseline_ssad.py
    • DeepSAD.py
    • baselines
      • ssad.py
      • isoforest.py
      • kde.py
      • ocsvm.py
      • SemiDGM.py
      • __init__.py
      • shallow_ssad
        • ssad_convex.py
        • __init__.py
    • __init__.py
    • main.py
    • baseline_kde.py
  • LICENSE
  • log
    • .gitkeep
  • imgs
    • .gitkeep
  • data
    • .gitkeep
  • README.md
  • requirements.txt

import torch
import matplotlib
matplotlib.use('Agg')  # or 'PS', 'PDF', 'SVG'

import matplotlib.pyplot as plt
import numpy as np
from torchvision.utils import make_grid


def plot_images_grid(x: torch.tensor, export_img, title: str = '', nrow=8, padding=2, normalize=False, pad_value=0):
    """Plot 4D Tensor of images of shape (B x C x H x W) as a grid."""

    grid = make_grid(x, nrow=nrow, padding=padding, normalize=normalize, pad_value=pad_value)
    npgrid = grid.cpu().numpy()

    plt.imshow(np.transpose(npgrid, (1, 2, 0)), interpolation='nearest')

    ax = plt.gca()
    ax.xaxis.set_visible(False)
    ax.yaxis.set_visible(False)

    if not (title == ''):
        plt.title(title)

    plt.savefig(export_img, bbox_inches='tight', pad_inches=0.1)
    plt.clf()