project_DIP
[DIP] Cephalogram Autolabeling03
jiwoolee
2021. 1. 11. 00:43
4) 최종 시행
앞선 과정을 통해 추출한 points가 정확한지 확인해보았다.
점의 최종 좌표를 기준으로 위로 15픽셀, 오른쪽으로 29픽셀을 잘라 영단어만 포함된 이미지를 만들고 다루기 쉬운 numpy array 형태로 리턴할 수 있도록 함수를 만들어주었다.
(크롭 이미지의 크기는 수차례 경험 후 최적의 사이즈를 도출해낸 것이다. 직접 수치를 조절하며 수정해야 한다.)
앞서 설계한 네트워크 모델을 학습시킨 데이터 중 최적의 weight를 load해 크롭된 이미지를 예측하도록 했다.
완성.
이미지를 예측한 결과를 csv파일로 만든다면 다음 단계에서 사용하기 용이할 것이다.
csv파일로 각 점들의 좌표를 저장해보자
최종 수행 코드: