Project /Bus Traffic Light Detection

19/04/15 진행 상황

murra 2019. 4. 16. 12:05

[19/04/15] 진행 상황

지난주의 계획대로 TL 전체가 아닌, Light 자체를 검출하는 모델을 만들어봤습니다.

Training dataset은 지난주에 사용했던 이미지들을(2088*1080) 두장으로 나누어(1080*1080) 2134개의 이미지를 썼습니다.

6천번 Training한 모델에 Test Video를 입력한 결과 아래 예시 이미지와 같이 나왔습니다.

Image

Image

Image

정확도를 측정하니 아래와 같이 나왔습니다.

class_id = 0, name = G, ap = 90.44 %
class_id = 1, name = Bus, ap = 99.15 %

for thresh = 0.25, precision = 0.94, recall = 0.99, F1-score = 0.96
for thresh = 0.25, TP = 665, FP = 43, FN = 6

mean average precision (mAP@0.50) = 0.947937, or 94.79 %