[data.transforms] Properly implement ElasticDeformation so it works with...
[data.transforms] Properly implement ElasticDeformation so it works with images or batches; Update test case for transforms; Create new test case for image_utils; Update models to reflect changes (partially addresses #5)
parent
2bd2a302
No related branches found
No related tags found
Showing
- src/ptbench/data/transforms.py 235 additions, 70 deletionssrc/ptbench/data/transforms.py
- src/ptbench/models/alexnet.py 1 addition, 6 deletionssrc/ptbench/models/alexnet.py
- src/ptbench/models/densenet.py 1 addition, 6 deletionssrc/ptbench/models/densenet.py
- src/ptbench/models/pasa.py 1 addition, 6 deletionssrc/ptbench/models/pasa.py
- tests/data/raw_with_elastic_deformation.png 0 additions, 0 deletionstests/data/raw_with_elastic_deformation.png
- tests/test_image_utils.py 56 additions, 0 deletionstests/test_image_utils.py
- tests/test_tranforms.py 8 additions, 51 deletionstests/test_tranforms.py
Loading
Please register or sign in to comment