[ptbench.data.datamodule] Implemented typing, added more logging, implemented...
[ptbench.data.datamodule] Implemented typing, added more logging, implemented sampler-balancing depending on sample class prevalence, parallelized dataset caching
parent
a69868bf
No related branches found
No related tags found
This commit is part of merge request !6. Comments created here will be created in the context of that merge request.
Showing
- src/ptbench/data/datamodule.py 274 additions, 150 deletionssrc/ptbench/data/datamodule.py
- src/ptbench/data/dataset.py 0 additions, 66 deletionssrc/ptbench/data/dataset.py
- src/ptbench/data/image_utils.py 11 additions, 9 deletionssrc/ptbench/data/image_utils.py
- src/ptbench/data/shenzhen/default.py 25 additions, 13 deletionssrc/ptbench/data/shenzhen/default.py
- src/ptbench/data/shenzhen/loader.py 105 additions, 0 deletionssrc/ptbench/data/shenzhen/loader.py
- src/ptbench/data/split.py 11 additions, 17 deletionssrc/ptbench/data/split.py
- src/ptbench/data/typing.py 75 additions, 0 deletionssrc/ptbench/data/typing.py
Loading
Please register or sign in to comment