python 機械学習　augmenter

augmentorで画像の水増しを行いたいのですが、水増ししたい階層の指定をどのようにしたらいいのかわからないので教えてください。

python
1import Augmentor
2
3p = Augmentor.Pipeline("/Users/human/Desktop/train/a")
4p.random_distortion(probability=0.5, grid_width=4, grid_height=4, magnitude=5)
5p.random_contrast(probability=0.5, min_factor=0.5, max_factor=1)
6p.random_color(probability=0.5, min_factor=0.8, max_factor=1)
7p.random_brightness(probability=0.5, min_factor=0.3, max_factor=1)
8p.rotate_random_90(probability=0.5)
9p.shear(probability=0.5, max_shear_left=10, max_shear_right=10)
10p.flip_left_right(probability=0.5)
11p.flip_top_bottom(probability=0.5)
12p.greyscale(probability=0.1)
13p.zoom_random(probability=0.3, percentage_area=0.8,
14              randomise_percentage_area=False)
15p.sample(40)
16
17p = Augmentor.Pipeline("/Users/human/Desktop/train/b")
18p.random_distortion(probability=0.5, grid_width=4, grid_height=4, magnitude=5)
19p.random_contrast(probability=0.5, min_factor=0.5, max_factor=1)
20p.random_color(probability=0.5, min_factor=0.8, max_factor=1)
21p.random_brightness(probability=0.5, min_factor=0.3, max_factor=1)
22p.rotate_random_90(probability=0.5)
23p.shear(probability=0.5, max_shear_left=10, max_shear_right=10)
24p.flip_left_right(probability=0.5)
25p.flip_top_bottom(probability=0.5)
26p.greyscale(probability=0.1)
27p.zoom_random(probability=0.3, percentage_area=0.8,
28              randomise_percentage_area=False)
29p.sample(40)
30
31p = Augmentor.Pipeline("/Users/human/Desktop/train/c")
32p.random_distortion(probability=0.5, grid_width=4, grid_height=4, magnitude=5)
33p.random_contrast(probability=0.5, min_factor=0.5, max_factor=1)
34p.random_color(probability=0.5, min_factor=0.8, max_factor=1)
35p.random_brightness(probability=0.5, min_factor=0.3, max_factor=1)
36p.rotate_random_90(probability=0.5)
37p.shear(probability=0.5, max_shear_left=10, max_shear_right=10)
38p.flip_left_right(probability=0.5)
39p.flip_top_bottom(probability=0.5)
40p.greyscale(probability=0.1)
41p.zoom_random(probability=0.3, percentage_area=0.8,
42              randomise_percentage_area=False)
43p.sample(40)

"/Users/human/Desktop/train/"のa.b.cのディレクトリを全て40枚ずつにしたいのですがパラメータを１ついじると全てのコードを変えなくてはならないので
"/Users/human/Desktop/train/ここにある全てのディレクトリ"といったコードに変えたいのですがどうしたらいいでしょうか？

宜しくお願い致します。