인프런 커뮤니티 질문&답변
작성한 질문수
Sequence의 __getitem__의 질문
작성
·
187
0
def __getitem__(self, index): 이렇게할때 여기서 index는 어디서 받아오는건가요? 함수 CnD_Dataset을 만들때는 __init__관련 값들만 넣어주고
(cnd_ds = CnD_Dataset(train_image_filenames, train_image_labels, batch_size=BATCH_SIZE, augmentor=cnd_augmentor, shuffle=False)
)index는 언급해주지않았는대, 어떤식으로 들어가서 나오는걸까요?
저는 cnd_ds[2]를 하게되면 [128:192]로 64개의 배치가 나온다고 이해하였는대 함수를 또 돌려보니 그렇지않아 질문드립니다.!
답변 1
0
안녕하십니까,
index는 호출하는 쪽에서 넣어 줍니다.
그리고 cnd_ds는 __getitem__(self, index)는 반환을 tuple을 합니다. images_batch와 labels_batch로 합니다. 때문에 cnd_ds[2] 를 하면 안되고, 3번째 batch iteration의 images_batch를 얻고 싶으면 cnd_ds[2][0]으로 labels_batch를 얻고 싶다면 cnd_ds[2][1]로 해야 합니다.
아래와 같이 수행해 보십시요.
cnd_ds[2][0].shape, cnd_ds[2][1].shape
cnd_ds를 모두 호출해 보기를 원하시면 아래와 같이 iteration을 적용해 주셔야 합니다. iteration 수행 시마다 index가 증가하면서 cnd_ds의 __getitem__(self, index) 를 호출하게 됩니다.
for index, (images_batch, labels_batch) in enumerate(iter(cnd_ds)):
print("index:", index, images_batch.shape, labels_batch.shape)
감사합니다.