Making use of LightningDataModule and simplification of data loading
Compare changes
Files
7- André Anjos authored
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: