Add upper limit to albumentations due to issue #57. Will try to remove the 'scale' dependency in the future.