인공지능/Pytorch 2

[Pytorch] torch.utils.data.DataLoader

앞선 글을 읽으시면 이해에 도움이 됩니다. 2023.02.25 - [인공지능/Pytorch] - [Pytorch] torch.utils.data.Dataset [Pytorch] torch.utils.data.Dataset 1. torch.utils.data.Dataset torch.utils.data의 하위에 있는 Dataset은 Pytorch에서 데이터를 불러오고, 전처리하는 클래스입니다. 따라서 사용자가 자신의 데이터셋을 만들어 Pytorch에서 사용할 수 있도록 해줍 hi-guten-tag.tistory.com 1. torch.utils.data.DataLoader torch.utils.data.dataloader는 PyTorch에서 제공하는 데이터 로더(data loader)입니다. 데이터 로더는..

[Pytorch] torch.utils.data.Dataset

1. torch.utils.data.Dataset torch.utils.data의 하위에 있는 Dataset은 Pytorch에서 데이터를 불러오고, 전처리하는 클래스입니다. 따라서 사용자가 자신의 데이터셋을 만들어 Pytorch에서 사용할 수 있도록 해줍니다. 이 클래스는 사용자 정의 데이터셋을 만들 수 있게 해주어 효율적인 데이터 로딩 및 전처리를 가능하게 합니다. 해당 클래스는 사용자가 직접 구현해야 합니다. torch.utils.data.Dataset을 상속받아 직접 구현합니다. 2. 파라미터 __init__(self, ...) : 해당 클래스의 인스턴스를 초기화 합니다. 이 메서드에서는 데이터셋에서 필요한 인자를 받습니다. 예를 들어, 데이터셋의 경로, 이미지 크기, 데이터 전처리 방법 등이 있을..