Augmentor Error

[Yung-zi]

I tried to reproduce this code. However, an error occurred in "img_aug.py". The error is listed as follows.

raise IndexError("There are no images in the pipeline. "
IndexError: There are no images in the pipeline. Add a directory using add_directory(), pointing it to a directory containing images.

[JuChanSeo]

I tried to reproduce this code. However, an error occurred in "img_aug.py". The error is listed as follows.

raise IndexError("There are no images in the pipeline. " IndexError: There are no images in the pipeline. Add a directory using add_directory(), pointing it to a directory containing images.

Hi, I had a similar problem and in my case, changing the path from relative path to absolute path when initializing "dataset_root_dir" solved this problem. Hope this helps!

[SinHanYang]

The above solution doesn't work for me.

[IkerSancho]

have fixed it like this:

import Augmentor
import os

def makedir(path):
    '''
    if path does not exist in the file system, create it
    '''
    if not os.path.exists(path):
        os.makedirs(path)
        os.chmod(path, 0o777)

datasets_root_dir = os.path.abspath('/home/capacity/ProtoPNet/datasets/cub200_cropped/')
dir = os.path.join(datasets_root_dir, 'train_cropped/')
target_dir = os.path.join(datasets_root_dir, 'train_cropped_augmented/')
makedir(target_dir)

folders = [os.path.join(dir, folder) for folder in next(os.walk(dir))[1]]
target_folders = [os.path.join(target_dir, folder) for folder in next(os.walk(dir))[1]]

#print("Folders to process:")
#for folder in folders:
#    images = os.listdir(folder)
#    print(f"Images found in {folder}: {len(images)}")
#print("\n")

#print("Target folders:")
#for folder in target_folders:
#    print(folder)
#print("\n")

for i in range(len(folders)):
    fd = folders[i]
    tfd = target_folders[i]
    makedir(tfd)

    '''
    img1 = os.listdir(fd)
    img2 = os.listdir(tfd)
    print(f"\nSource directory:{fd}\t n_images :: {len(img1)}")
    print(f"Destination directory:{tfd}\t n_images :: {len(img2)}\n")
    '''

    # Check if there are images in the folder
    images = os.listdir(fd)
    if len(images) == 0:
        print(f"No images found in folder: {fd}")
        continue

    print(f"Processing folder: {fd} with {len(images)} images")

    # Apply augmentation pipeline (rotation)
    try:
        p = Augmentor.Pipeline(source_directory=fd, output_directory=tfd)
        p.rotate(probability=1, max_left_rotation=15, max_right_rotation=15)
        p.flip_left_right(probability=0.5)
        for _ in range(10):  # Process 10 batches
            p.process()
        del p
    except Exception as e:
        print(f"Error processing rotation in folder {fd}: {e}")

    # Apply skew pipeline
    try:
        p = Augmentor.Pipeline(source_directory=fd, output_directory=tfd)
        p.skew(probability=1, magnitude=0.2)  # max 45 degrees
        p.flip_left_right(probability=0.5)
        for _ in range(10):  # Process 10 batches
            p.process()
        del p
    except Exception as e:
        print(f"Error processing skew in folder {fd}: {e}")

    # Apply shear pipeline
    try:
        p = Augmentor.Pipeline(source_directory=fd, output_directory=tfd)
        p.shear(probability=1, max_shear_left=10, max_shear_right=10)
        p.flip_left_right(probability=0.5)
        for _ in range(10):  # Process 10 batches
            p.process()
        del p
    except Exception as e:
        print(f"Error processing shear in folder {fd}: {e}")

    # Uncomment to process random distortion if needed
    #try:
    #    p = Augmentor.Pipeline(source_directory=fd, output_directory=tfd)
    #    p.random_distortion(probability=1.0, grid_width=10, grid_height=10, magnitude=5)
    #    p.flip_left_right(probability=0.5)
    #    for _ in range(10):
    #        p.process()
    #    del p
    #except Exception as e:
    #    print(f"Error processing random distortion in folder {fd}: {e}")