app/DataProcessor/SingleImageProcessor.py

import torch

from pathlib import Path
from app.DataProcessor.ImageProcessor import ImageProcessor

class SingleImageProcessor(ImageProcessor):
    def process_input_data(self, image_file : Path | str):
        img = self._get_img_tensor(Path(image_file[0]))
        img = img[None, None, :].repeat(self.NUM_PROPOSALS, 1, 1, 1, 1)
        img_id = torch.tensor([[0]], device=self._device).repeat(self.NUM_PROPOSALS, 1)
        return {
            "imgs" : img,
            "img_id" : img_id
        }