[TIL]KDT팀프로젝트2021/03/12

opencv에도 눈코입 디텍션하는게 있는데, 전에 학부 과제 했을때 써봤지만 속도나 성능 측면에서 만족스럽지 못했던 걸로 기억한다(특히 극악의 fps) 일단 Detectron2로 계속 해보려고 한다.
모델pth 파일이 있으면 이전에 진행하던 학습을 이어서 할 수 있다는 사실을 알았다.
훈련한 pth파일을 다시 훈련하는 과정을 반복하다가(800 iteration -> 2000 iteration, 101층이라 엄청 오래걸림) 답답해서 코랩 프로를 결재했다.(다음주에 깨달은 사실이지만 런타임이 끊기면 이어서 학습이 안되는 문제가 있었다. 뻘짓했던거임)
train/val로 나눠서 학습하고 plot을 도출해 validation score와 train score를 비교해서 오버피팅을 방지할 수 있는 방법을 시도하고 있다.
다음은 참고한 링크들.
https://medium.com/@apofeniaco/training-on-detectron2-with-a-validation-set-and-plot-loss-on-it-to-avoid-overfitting-6449418fbf4e
https://gist.github.com/ortegatron/c0dad15e49c2b74de8bb09a5615d9f6b#file-plottogether-py
https://github.com/sherrychen127/face_mask_detector/tree/e48d4df14619fdeb0ba53b5283833c28ef8ccc24