Merge branch 'add-datamodule-andre' into 'add-datamodule'
Reviewed DataModule design+docs+types See merge request biosignal/software/ptbench!7
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
- doc/references.rst 3 additions, 2 deletionsdoc/references.rst
- pyproject.toml 30 additions, 30 deletionspyproject.toml
- src/ptbench/configs/models/alexnet.py 15 additions, 4 deletionssrc/ptbench/configs/models/alexnet.py
- src/ptbench/configs/models/alexnet_pretrained.py 16 additions, 5 deletionssrc/ptbench/configs/models/alexnet_pretrained.py
- src/ptbench/configs/models/densenet.py 15 additions, 5 deletionssrc/ptbench/configs/models/densenet.py
- src/ptbench/configs/models/densenet_pretrained.py 15 additions, 5 deletionssrc/ptbench/configs/models/densenet_pretrained.py
- src/ptbench/configs/models/pasa.py 12 additions, 16 deletionssrc/ptbench/configs/models/pasa.py
- src/ptbench/data/base_datamodule.py 0 additions, 196 deletionssrc/ptbench/data/base_datamodule.py
- src/ptbench/data/datamodule.py 722 additions, 0 deletionssrc/ptbench/data/datamodule.py
- src/ptbench/data/dataset.py 0 additions, 586 deletionssrc/ptbench/data/dataset.py
- src/ptbench/data/image_utils.py 100 additions, 0 deletionssrc/ptbench/data/image_utils.py
- src/ptbench/data/padchest/__init__.py 1 addition, 1 deletionsrc/ptbench/data/padchest/__init__.py
- src/ptbench/data/shenzhen/__init__.py 0 additions, 19 deletionssrc/ptbench/data/shenzhen/__init__.py
- src/ptbench/data/shenzhen/default.py 35 additions, 16 deletionssrc/ptbench/data/shenzhen/default.py
- src/ptbench/data/shenzhen/loader.py 105 additions, 0 deletionssrc/ptbench/data/shenzhen/loader.py
- src/ptbench/data/shenzhen/rgb.py 30 additions, 71 deletionssrc/ptbench/data/shenzhen/rgb.py
- src/ptbench/data/shenzhen/utils.py 0 additions, 114 deletionssrc/ptbench/data/shenzhen/utils.py
- src/ptbench/data/split.py 252 additions, 0 deletionssrc/ptbench/data/split.py
- src/ptbench/data/transforms.py 23 additions, 41 deletionssrc/ptbench/data/transforms.py
- src/ptbench/data/typing.py 75 additions, 0 deletionssrc/ptbench/data/typing.py
Loading
Please register or sign in to comment