pytorch2 [Pytorch] 이미지 데이터셋 만들기 나만의 이미지 데이터셋 만들기 방법이 여기 잘 설명되어 있다 https://data-panic.tistory.com/13 [PyTorch] 4-1. 나만의 이미지 데이터셋 만들기 이번 포스트에서는 이전 시간에 다루었던 DataLoader를 가지고 나만의 이미지 데이터 셋을 만들어보는 실습을 진행해보겠습니다. 1. 이미지 데이터 파일 만들기 이미지 데이터 셋을 만들기 위해서 data-panic.tistory.com 2021. 10. 17. [Pytorch] tensor에서 특정 조건에 맞는 원소의 인덱스 구하기 Pytorch 프로그램을 하다 보면 tensor에서 특정 조건에 맞는 원소의 인덱스를 구해야 하는 경우가 많다 이런 경우 다음과 같이 하면 된다. import torch a = torch.tensor([1,2,3,4]) b = torch.tensor([0,4,3,5]) # 두 개의 tensor에서 다른 부분의 인덱스 구하기 idx = ((a-b)!=0).nonzero().flatten() print(idx.tolist()) # [0,1,3] # a 원소가 b 원소보다 큰 경우의 인덱스 구하기 idx = ((a-b)>0).nonzero().flatten() print(idx.tolist()) # [0] # a의 최대값과 최대값의 인덱스 구하기 print('Max: ', a.max(dim=0)[0].toli.. 2020. 5. 15. 이전 1 다음
