YOLO-Object-Detection / test_cases / get_class_prediction_output.pt