efficientDet을 이용한 detection, deeplearning

저번에 문제를 해결하고, 학습을 하니 위 그림과 같이 제대로 학습이 된 걸 확인할 수 있었다.

사용한 코드는

https://github.com/wangermeng2021/EfficientDet-tensorflow2

GitHub - wangermeng2021/EfficientDet-tensorflow2: A Tensorflow2.x implementation of EfficientDet

잘 돌아간다.

여기서 d2를 이용했다.

d4만 돼도 메모리 에러가 뜬다. 25GB 메모리도 부족한가보다. 베치 사이즈 1로 했음에도 불구하고...

새로운 문제

문제는 테스트 데이터는 한 이미지에 객체가 여러개 존재한다는 것

단일 이미지로 단일 객체로 학습한 모델로는 여러 객체를 탐지하지 못한다.

잘라서 테스트하면?? 그래도 제대로 탐지하지 못함을 확인함

 

멀티레이블로 구성된 이미지를 100~200장 정도만 학습하고 해보자.