python setup.py install
# It's important that you do this first import torch from pytorch_image_ops import color_transform, spatial_transform