Imageomics / Imageomics/hpc-inference

Investigate GPU-Accelerated Image Preprocessing

Open
#5 1 comment 0 reactions 1 assignee View on GitHub

@NetZissou is already working on this.

Since Jul 16, 2025.

exploration
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

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.