Dev batch align module (#5401)
* add image_batch_align module
* add testcase
Co-authored-by:
oneflow-ci-bot <69100618+oneflow-ci-bot@users.noreply.github.com>
Showing
- oneflow/python/nn/modules/dataset.py 19 additions, 0 deletionsoneflow/python/nn/modules/dataset.py
- oneflow/python/test/modules/test_image_batch_align.py 97 additions, 0 deletionsoneflow/python/test/modules/test_image_batch_align.py
- oneflow/user/kernels/image_batch_align_kernel.cpp 20 additions, 10 deletionsoneflow/user/kernels/image_batch_align_kernel.cpp
Please register or sign in to comment