728x90
저번에 문제를 해결하고, 학습을 하니 위 그림과 같이 제대로 학습이 된 걸 확인할 수 있었다.
사용한 코드는
https://github.com/wangermeng2021/EfficientDet-tensorflow2
GitHub - wangermeng2021/EfficientDet-tensorflow2: A Tensorflow2.x implementation of EfficientDet
A Tensorflow2.x implementation of EfficientDet. Contribute to wangermeng2021/EfficientDet-tensorflow2 development by creating an account on GitHub.
github.com
잘 돌아간다.
여기서 d2를 이용했다.
d4만 돼도 메모리 에러가 뜬다. 25GB 메모리도 부족한가보다. 베치 사이즈 1로 했음에도 불구하고...
새로운 문제
문제는 테스트 데이터는 한 이미지에 객체가 여러개 존재한다는 것
단일 이미지로 단일 객체로 학습한 모델로는 여러 객체를 탐지하지 못한다.
잘라서 테스트하면?? 그래도 제대로 탐지하지 못함을 확인함
멀티레이블로 구성된 이미지를 100~200장 정도만 학습하고 해보자.
'Data-science > deep learning' 카테고리의 다른 글
| [torch] RuntimeError("{} is a zip archive 에러 해결 방법 (0) | 2021.11.08 |
|---|---|
| 배경 차이가 뚜렷한 이미지에서, cv2.connectedComponetsWithStats 활용하여 객체 분리하고 자르기 + efficientDet (0) | 2021.08.30 |
| [detection] 이미지 labeling이 이상하다 싶지만 정상일 때 해결방법. PIL Image rotated 회전시키는 녀석. (PIL image 쓸데 없이 고성능) (0) | 2021.08.28 |
| [딥러닝] Graph2Vec 4가지 특징 (0) | 2021.05.04 |
| [pytorch] torch.utils.data.DataLoader 이용시 파일 경로 출력 (0) | 2021.03.13 |
