YOLO-Object-Detection / test_cases / get_regression_output.pt