如何使用ITK色彩映射转换文件夹中的多张图片？

#include <iostream>
#include <string>
#include <itkImage.h>
#include <itkImageFileReader.h>
#include <itkImageFileWriter.h>
#include <itkColorMapImageFilter.h>

int main()
{
    std::string inputFolder = "path/to/input/folder/";
    std::string outputFolder = "path/to/output/folder/";

    // 构造输入图像读取器
    using ImageType = itk::Image<unsigned char, 2>;
    using ReaderType = itk::ImageFileReader<ImageType>;
    ReaderType::Pointer reader = ReaderType::New();

    // 构造色彩映射转换器
    using ColorMapFilterType = itk::ColorMapImageFilter<ImageType, ImageType>;
    ColorMapFilterType::Pointer colorMapFilter = ColorMapFilterType::New();

    // 构造输出图像写入器
    using WriterType = itk::ImageFileWriter<ImageType>;
    WriterType::Pointer writer = WriterType::New();

    // 遍历文件夹中的图片
    // 假设文件名为image1.png、image2.png、image3.png...
    for (int i = 1; i <= 3; ++i)
    {
        std::string inputFilename = inputFolder + "image" + std::to_string(i) + ".png";
        std::string outputFilename = outputFolder + "image" + std::to_string(i) + "_color_mapped.png";

        // 读取输入图像
        reader->SetFileName(inputFilename);
        reader->Update();

        // 进行色彩映射转换
        colorMapFilter->SetInput(reader->GetOutput());
        // 设置映射规则，例如使用灰度映射
        colorMapFilter->SetColormap(ColorMapFilterType::Grey);
        colorMapFilter->Update();

        // 保存转换后的图像
        writer->SetFileName(outputFilename);
        writer->SetInput(colorMapFilter->GetOutput());
        writer->Update();
    }

    return 0;
}