inference 평가 및 시각화

[jerry-ryu]

Overview

• 현재 예측이 validation만 이루어지고 있으므로, 전체 데이터셋의 pred와 gt 값을 비교하기 위해 inference code가 필요하다

Todo

inference code

• [x] inference code - ufo 생성
• [x] inference code - 평가
• [x] inference code & gt 시각화

Additional Context

• fifty one을 사용하여 pred와 gt의 직접적 비교가 가능하도록 하면 편할 것 같다.

[jerry-ryu]

inference 평가 및 시각화 사용 방법

1. inference_viz.py 실행

   --data_dir == 데이터 폴더 위치 --json_dir == json 파일 위치 --model_dir == 모델 pth 파일 위치 --output_dir == 출력 파일 위치

2. inf가 완료되면. output_dir에 result.txt와 output.json이 생긴다. result.txt에는 평가지표 점수가 저장되어있다.

3. notebook/FiftyOne.ipynb pred_gt 비교에 이미지 파일 경로와 output.json을 넣고 fifty one을 실행하면

gt와 pred를 한번에 볼 수 있다.