728x90
배치 사이즈가 0번째 축이라고 가정했을 때
(5, 256, 256, 3) 이라는 이미지가 있고 (B, W, H, C) 이를
(5, 3, 256, 256) 차원으로 변경하려면
np.rollaxis(3, 1)
3번 축을 1번 축까지 굴림.
'Data handling' 카테고리의 다른 글
[pandas] 특정 칼럼이 문자열로 이루어졌을 경우, 특정 조건을 만족하는 row만 select 하기 (0) | 2021.07.15 |
---|---|
[numpy] list of array -> numpy array 변환 (0) | 2021.07.05 |
대용량 numpy 파일 text로 쓰기 (0) | 2021.04.05 |
Pandas error, csv를 읽고 쓸 때 발생하는 유니코드 에러 (0) | 2021.02.21 |
json 파일 저장시 에러 (0) | 2021.01.14 |