Warboy-vision-models 활용 시, batch size 조절 방법이 궁금합니다

한국어 (Korean)
1

안녕하세요.

Warboy로 추론 성능을 시험 중입니다.

warboy-vision-models으로 아래 명령을 수행 시,
warboy-vision model-performance --config_file tutorials/cfg/yolov8n.yaml

  1. batch_size를 변경할 수 있는 방법이 있을지요?
  • 현재는 1로 고정되어 있는 것 같습니다.
  1. conf_thres, iou_thres 의 값을 yaml 파일에 기재하여도, src/test_scenarios/utils.py 에 명시된 값이 사용되는 것이 맞는지요?

관련하여, 코드 수정이 필요할지, 어느 부분을 중점적으로 수정해야할 지 조언 부탁드립니다.
감사합니다.

2

안녕하세요, 퓨리오사에이아이 이지수입니다.

  1. batch 처리의 경우, 실험해보았을 때 batch 1인 경우가 가장 효과적이어서 따로 코드 정리를 진행하진 않았었습니다. 다만, 실험을 원하신다면 제가 실험할 때 사용하였던 코드는 링크의 레포지토리의 브랜치에서 확인 가능하십니다.
  2. 말씀해주신 부분 같은 경우에는 기존에 src/warboy/utils/process_pipeline.py 의 Engine class의 정의 부분에서 정의된 default 값으로 설정되고 있는 코드 오류를 발견하였습니다. 감사합니다. 금일 중으로 수정된 코드 main 브랜치에 머지해두도록 하겠습니다.

감사합니다.

3

conf thres, iou thres 관련 수정된 코드 메인 브랜치에 올라갔습니다.

제보 감사드립니다.