[AITech][Final Project][P stage] 20220530 - 최종 프로젝트 11일차

최종 프로젝트 11일차

주말동안 pytorch로 코드 리팩토링하는데 성공했다.

크게 model, preprocessing(data transform), input format(ndarray VS PIL.Image) 등을 바꿔주었다. 성능이 유지되지 않는 모습을 보여 바꾸는게 쉽지는 않았지만, 최종적으로 성공하며 tensorflow와 pytorch 코드 간 관계성에 대해 조금이나마 이해를 했던 것 같다.

해당 작업이 마무리 된 후 처리 속도를 올리기 위해 추가적인 코드 작성을 진행했다.

기존 코드에서는 1장씩 이미지를 처리했었는데, 이를 N 장 단위로 모아서 배치 단위로 처리하도록 코드를 작성하였다. 그 과정에서 clustering의 성능이 조금씩 변하는 현상이 생겨, 이를 해결하기 위해 노력 중이다. (속도를 높이는 것은 성능이 어느정도 유지가 될 때 의미있는 것이기에…)

우선 설정 가능한 하이퍼파라미터들을 바꿔주며 실험을 통해 최적의 파라미터 세팅을 찾는 과정을 수행할 것 같다.

결론

• 배치 단위 처리 시 성능의 유지

참고 자료