Imageomics / Imageomics/hpc-inference
Investigate GPU-Accelerated Image Preprocessing
@NetZissou is already working on this.
Since Jul 16, 2025.
- Dominant language
- Python
- Stars
- 0
- Forks
- 0
- PR merge metrics
- No merged PRs in 30d
Description
# Overview
Currently, image preprocessing (such as resize and normalization) is performed on the CPU, which can become a bottleneck for high-throughput pipelines. Some libraries allow moving these steps onto the GPU, which can dramatically speed up batch inference if images are already loaded as tensors or in GPU memory.
# Relevant Libraries
- [torchvision.transforms.v2](https://pytorch.org/vision/stable/transforms.html#transforms-on-torch-tensors)
- [Kornia](https://kornia.org/)
- [NVIDIA DALI](https://docs.nvidia.com/deeplearning/dali/user-guide/docs/index.html)
# Potential Benefits
- Reduce CPU-GPU bottleneck during image preprocessing
- Increase throughput for large-scale inference tasks
# Action Items
- Explore feasibility of moving preprocessing steps onto the GPU using the above libraries.
- Benchmark performance vs. current CPU-based approach.
- Assess ease of integration with the current pipeline
Contributor guide
No contributing guide indexed for this repository
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Assessment
This issue has not been assessed yet.