YOLO v5 - 자동차 번호판(License Plate) 인식 - 3. Vehicles Detection 하기

학습이 완료 되면 yolov5-master 디렉토리의 runs/train/vehicls_yolov5s_results를 확인 합니다.

 

val_batch2_pred의 내용입니다. 모델이 Validation set을 predict한 내용인데요 학습이 잘된것을 확인 할수 있습니다.

역시 Small 모델이라서 Large/eXtra Large보다 Detection 성능이 조금 떨어지지만 인식 속도는 빠릅니다.

 

Detetion할때 weight 파일은 best.pt을 쓸 예정입니다.

 

테스트할 이미지 파일이나 동영상 파일을 구해옵니다.

 

yolov5-master 디렉토리로 가서 아래의 명령어를 입력합니다. 테스트할 동영상은 road1.mp4 입니다.

python detect.py --source ./test_data/movies/road1.mp4 --weights ./runs/train/vehicles/_yolov5s_results/weights/best.pt

 

Detetion 중입니다. Detection 중간 과정을 나타냅니다.

 

yolov5-master/runs/detect로 가시면 Detetion 결과인 exp 디렉토리가 생성된 것을 볼 수 있습니다.

 

Detection 결과가 담긴 동영상을 플레이 합니다.

 

자동차가 Detection 되는 모습을 확인 할 수 있습니다.

다음은 Public dataset 해봤으니 Custom dataset을 생성하는 방법을 알아 보겠습니다.

Detection 결과