references/optical_flow/utils.py
references/segmentation/utils.py

references/optical_flow/train.py
references/segmentation/train.py