components/dataprep/dataprep.yaml

$schema: https://azuremlschemas.azureedge.net/latest/commandComponent.schema.json

name: data_prep_image_resize_and_augment
version: 0.1.0
display_name: Data Preparation - Image Resizing and Augmentation
is_deterministic: True
type: command
description: Reads a dataset of Alzheimer's images, resizes them to 64x64, and balances classes by augmenting smaller classes.

environment: azureml:aml-preprocessing-cli:0.1.0
resources:
  instance_count: 1

code: ./code
command: 'python dataprep.py --data ${{inputs.data}} --output_data ${{outputs.output_data}}'

inputs:
  data:
    type: uri_folder
    optional: False
    description: Path to the input dataset folder containing class subfolders.

outputs:
  output_data:
    type: uri_folder
    description: Path to the output folder where resized and augmented images will be saved.