YOLO-Object-Detection / test_cases / get_class_prediction_input.pt