YOLO 모델 서빙 코드 성능 향상 경험
이 글은 알고리즘 팀에서 작성한 모델 파일을 통해 이미지를 추론하여 그 결과를 가공하고 DB에 적재하는 업무를 하던 중 경험한 내용을 정리한 글입니다.먼저 알고리즘 팀으로부터 모델 파일과 추론 결과값 형식을 제공받은 후 서빙 코드를 작성했습니다.def inference_on_gpu(device_id: int, image_list: List[str]) -> None: try: # 모델 로드 model = load_model(model_path=model_path, device_id=device_id, save_dir=save_dir, save_folder="run") for jpg_filename in image_list: # 이미지 하나씩 모..