[알약 분류] 2. 알약 분류

 

 

 

 

 

---

알약분류

탐지 결과는 만족한다고 합니다. 이번에는 탐지한 알약이 어떤 알약인지 분류할 수 있냐고 의뢰가 생겼습니다.

 

알약의 색과 크기가 비슷한게 많아서 정확하게 분류하는 것은 거의 불가능하지만, 비슷한 알약끼리 리스트로 분류가 가능하다고 했습니다.

예를 들어 빨간색에 길쭉한 알약, 검정색의 동그란 알약 이런식으로 분류가 가능하다고 했습니다.

 

일단, 샘플 데이터로는 알약 종류가 8종류라 정확하게 분류할 수 있는지 확인해보려고 합니다.

 

 

---

라벨링

다음과 같이 클래스를 세분화 했습니다.

---

 

Train

데이터셋을 다운로드하고 Train합니다.

 

 

 

---

 

Test

 

 

Test 분류는 되지만, 불안정한 모습을 볼 수 있습니다.

 

 

 

---