pytorch/sagemakercv/data/transforms/transforms.py
pytorch/sagemakercv/utils/visualize.py