-
- Downloads
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
Checking pipeline status
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