분류 전체보기 썸네일형 리스트형 [PyTorch] ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1, 32]) Traceback (most recent call last): File "train.py", line 193, in g_loss = torch.mean(torch.abs(netD(gen_imgs) - gen_imgs)) File "/home/cvmi-koo/.local/lib/python3.8/site-packages/torch/nn/modules/module.py", line 889, in _call_impl result = self.forward(*input, **kwargs) File "/home/cvmi-koo/code/Face-Generator/libs/models.py", line 77, in forward self.bn1(out) File "/home/cvmi-koo/.local/lib/py.. Python에서의 이미지 불러오기 속도 비교(PIL, cv2, imageio, tf) Python에서 이미지를 읽는 방법은 여러가지가 있습니다. 컴퓨터비전을 위한 머신러닝, 딥러닝에선 이미지를 불러와 직접 다루는 일이 많습니다. 특히 이미지로 이루어진 커다란 데이터셋의 경우, 이미지를 불러오는 속도조차 신경 쓰일 수 있습니다. 이번 글에서는 4가지의 파이썬 외부 패키지의 이미지 로드 속도를 비교해보겠습니다. 아래 코드는 Windows 10 PC에서 실행하였습니다. import os import time import numpy as np import cv2 import PIL import imageio import tensorflow as tf print('PIL:', PIL.__version__) print('cv2:', cv2.__version__) print('imageio:', im.. 이전 1 다음
