PyTorch를 이용한 3채널 이미지 평규과 표준 편차 구하기

main의 files 배열에 평균과 표준 편차를 구할 이미지들의 경로를 전달하기만 하면 평균과 표준 편차를 torch.tensor() 형태로 반환해 준다. from typing import List import torch from PIL import Image from torchvision.transforms import transforms from glob import glob class NormStdDataSet(object): def __init__(self, dataset_list: List[str]): super().__init__() self.dataset_list = dataset_list self.length = len(self.dataset_list) self.transform = tra..