Please see the colab notebook used for training the model. Python scripts depend on the affinity estimator from https://github.com/xiaolonw/UVC-1.git