Files
shape_ai/training/generate_shapes.py