Python/시각화
[CV] tensor 이미지에서 엣지 추출하기
from torch.utils.data import DataLoader의 DataLoader로 저장된 tensor image 엣지 추출하기 import cv2 img, label = train_dataset[0] #numpy로 바꿔서 진행해야함 img = img.detach().cpu().numpy() # tensor -> numpy # [C,H,W] -> [H,W,C] img = np.transpose(img, (1, 2, 0)) #color로 나오게 수치 조정 img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB) img = img.astype(np.uint8).copy() #엣지 추출 edge_img = cv2.Canny(img, 100, 500) #안해도 되지만 이미지가 작다..
