PyTorch 특정 레이어 시각화 방법
이미지를 로드하고 전처리하는 코드입니다.
import cv2
import matplotlib.pyplot as plt
%matplotlib inline
# 자신의 이미지 파일 경로로 변경 가능합니다
img_path = 'images/udacity_sdc.png'
# 컬러 이미지 로드
bgr_img = cv2.imread(img_path)
# 그레이스케일로 변환
gray_img = cv2.cvtColor(bgr_img, cv2.COLOR_BGR2GRAY)
# 값을 [0,1] 범위로 정규화
gray_im ...
5월 21일 02:42에 게시됨