Created common DataModule for Shenzhen dataset
Showing
- src/ptbench/data/base_datamodule.py 4 additions, 0 deletionssrc/ptbench/data/base_datamodule.py
- src/ptbench/data/shenzhen/default.py 9 additions, 86 deletionssrc/ptbench/data/shenzhen/default.py
- src/ptbench/data/shenzhen/utils.py 114 additions, 0 deletionssrc/ptbench/data/shenzhen/utils.py
- src/ptbench/scripts/train.py 1 addition, 1 deletionsrc/ptbench/scripts/train.py
Loading
Please register or sign in to comment