You could pass the transforms.Compose as the transform argument to the STLData, if you would like to use it for this dataset.
Note that torchvision.transforms work on PIL.Images by default (in the nightly more transformation can also be applied to tensors directly) so you might need to transform the numpy arrays to PIL.Images first.