Detection Task에서 collate_fn의 입·출력 구조와 default_collate의 한계
Detection task를 위한 collate_fnDetection 테스크의 경우, DataLoader에서 collate_fn을 일반적으로 다음과 같이 변경해줘야 함.def collate_fn(batch): # batch: [(img, target), (img, target), ...] imgs, targets = zip(*batch) return list(imgs), list(targets)loader = DataLoader( train_ds, batch_size=2, shuffle=True, collate_fn=collate_fn,)detection task의 경우, 각 image 별로 가지고 있는 object의 수가 다를 수 있음.그 결과 targets..
2025. 12. 15.
