src/pipeline/pipeline_steps.py

# ----------------------------------------------------------------------
# IMPORTS
# ----------------------------------------------------------------------
from src.processing.image_download.image_download import image_download
from src.processing.remove_background.remove_background import remove_background
from src.processing.bounding_box.bounding_box import bounding_box
from src.processing.cropping_padding.cropping_padding import cropping_padding
from src.processing.return_images.return_images import return_images

# ----------------------------------------------------------------------
# PIPELINE STEPS
# ----------------------------------------------------------------------
PIPELINE_STEPS = [
    image_download,
    remove_background,
    bounding_box,
    cropping_padding,
    return_images
]