Jetson에서 같은 숫자 배열도 채널 축을 어디에 두는지에 따라 해석이 달라진다.
import numpy as np
x = np.zeros((1, 3, 224, 224), dtype=np.float32)
y = x.transpose(0, 2, 3, 1)
print(x.shape, y.shape)(1,3,224,224)를 (1,224,224,3)으로 전치한다. reshape로 모양만 바꾸는 것과 데이터 축을 교환하는 것은 다르다.
Python 코드를 example.py로 저장해 python3 example.py로 실행한다. import하는 모듈과 예제에 지정한 입력 파일이 준비돼 있어야 한다.
참고: NumPy transpose
답글 달기