YOLO-Object-Detection / test_cases / get_regression_input.pt