[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)

#안해도 되지만 이미지가 작다면 키우자
edge_img = cv2.resize(edge_img, dsize=(0,0), fx=10, fy=10, interpolation=cv2.INTER_LINEAR)

cv2.imshow('color image', edge_img)
cv2.waitKey()