This version artificially expands the data using jittered versions of the original images.
This should correctly cluster the images by finding the best random feature among a list of them, which is dependent on tree depth. This also avoids problems where the cluster is too small, and some branch cannot reach leaves of depth d=tree_depth.