This merge request adds LightningDataModule to better organize the code and make better use of lightning's features. This centralizes common tasks such as DataLoader creation and application of transforms into a base class to be inherited from.
Data loading was also simplified by removing custom Sample
classes and maker
functions, and the addition of RuntimeDataset
and CachedDataset
.
Remaining tasks:
tbx11k
, protocol v1
(uses original dataset organisation)tbx11k
, protocol v2
(uses original dataset organisation)montgomery-shenzhen
montgomery-shenzhen-indian
montgomery-shenzhen-indian-tbx11k-v1
montgomery-shenzhen-indian-tbx11k-v2
montgomery-shenzhen-indian-padchest
nih-cxr14
(n.b.: multi-class dataset, radiological findings)nih-cxr14-padchest
Addresses the following issues: