torch.numel torch.numel input tensor에 대해 element의 총 개수를 return >>> a = torch.randn(1, 2, 3, 4, 5) >>> torch.numel(a) 120 >>> a = torch.zeros(4,4) >>> torch.numel(a) 16

Custom mini batch Loader로 원하는 데이터 mini batch에 강제하기 현재 축관 제조공정 중 결함을 감지하는 Object detection Task에서 데이터 셋 3만장 중 결함이 있는 데이터는 3%에 불과합니다. 하지만 학습환경상 최대 설정 batch size가 8이었고, mini batch 안에 결함이 들어갈 이미지가 들어갈 확률은 극도로 작아 결함데이터에 대해 학습하지 못할 가능성이 높았습니다. 따라서 custom mini batch loader를 만들어 이 문제를 해결하려 했습니다. 1) Pytorch 구현 방식 파이토치에서 기본적으로 데이터로더를 쓸 때 불러와지는 것은 Dataset RandomSampler BatchSampler DataLoader 이 4가지 입니다. py..