Skip to content

Instantly share code, notes, and snippets.

View Rm1n90's full-sized avatar

Armin Mehri Rm1n90

  • Computer Vision Center
  • Barcelona, Spain
View GitHub Profile
Rm1n90 /
Last active May 30, 2019 22:58
Computing the mean and std
computing the mean and the standard deviation per channel of any datasets with PyTorch
import numpy as np
import torch
import torch.nn as nn
import torchvision
from import DataLoader
from import Dataset