/COCO_dataset_YOLOv5_test

YOLO v5 object detection test using COCO dataset

Primary LanguageJupyter NotebookGNU Lesser General Public License v3.0LGPL-3.0

COCO_dataset_YOLOv5_test

YOLO v5 object detection test using COCO dataset

데이터셋 : COCO(comman object in context) : 마이크로 소프트에서 만든 객체탐지에 사용되는 데이터셋

train 이미지 수 : 116408, validation 이미지 수 : 5000

모델 : YOLO v5

---- 8.11. first test ----

  • batch : 16
  • epochs: 1
  • imagesize: 640
  • 1 에폭 학습시키는데 구글 코랩 gpu 기준 1시간정도 소요됨

---- 8.18. additional tests ----

  • epochs: 2, confidence : 0.5

---- 8.19. additional tests ----

  • epochs: 3, confidence : 0.25

---- 8.20. additional tests ----

  • epochs: 30, confidence : 0.25

======= test results ==============

---------------------- 8.22. test video results 30 에폭

Video, 동영상 객체탐지 테스트

(썸네일 클릭시 영상 링크로 이동)

(Click the thumbnail to move the video link)

공항 객체 탐지 결과 동영상 (사람, 핸드백)

공항 객체 탐지 동영상

비오는 날 객체 탐지 결과 동영상 ( 자동차, 사람, 우산)

비오는 날 객체 탐지 동영상

폭동 객체 탐지 결과 동영상 (사람)

폭동 객체 탐지 동영상

서울 거리 객체 탐지 결과 동영상 (사람)

서울 거리 객체 탐지 동영상

백조 객체 탐지 결과 동영상 (bird)

백조 객체 탐지 동영상

싱가포르 거리 객체 탐지 결과 동영상 (사람, 자동차, 버스, 신호등)

싱가포르 거리 객체탐지 동영상

말 객체 탐지 결과 동영상 : 클래스 오분류 사례 (cow, dog)

말 객체 탐지 동영상

잉어 객체 탐지 결과 동영상 : 클래스 오분류 사례 (hotdog, carrot)

잉어 객체탐지 동영상

사슴 무리 객체 탐지 결과 동영상 : 클래스 오분류 사례 (horse)

사슴 무리 객체 탐지 동영상

사슴 이동 객체 탐지 결과 동영상 : 클래스 오분류 사례 (horse)

사슴 이동 객체 탐지 동영상

---------------------- 8.20. test results 30 epochs

Video, 동영상 객체탐지 테스트

(썸네일 클릭시 영상 링크로 이동)

(Click the thumbnail to move the video link)

해변 객체 탐지 결과 동영상

해변 객체탐지 결과 동영상

조류 객체 탐지 결과 동영상

조류 객체탐지 결과 동영상

고양이 객체 탐지 결과 동영상

고양이 객체탐지 결과 동영상

EnlightenGAN으로 조도개선 img 객체탐지 테스트

dst3_screenshot_02 08 2021 dst5_screenshot_02 08 2021 dst6_screenshot_02 08 2021 dst12_screenshot_11 08 2021 elgan

gray scale img 객체탐지 테스트

grayimage1 grayimage2 grayimage3 grayimage4 grayimage6 grayimage7 grayimage8 grayimage9 grayimage5

---------------------- 8.19. test results 3 epochs

EnlightenGAN으로 조도개선 img 객체탐지 테스트

dst3_screenshot_02 08 2021 dst5_screenshot_02 08 2021 dst12_screenshot_11 08 2021 KakaoTalk_20210720_213727400

gray scale img 객체탐지 테스트

KakaoTalk_20210811_144832280 KakaoTalk_20210811_144832280_01 KakaoTalk_20210811_144832280_02 KakaoTalk_20210811_144832280_03 KakaoTalk_20210811_144832280_04 KakaoTalk_20210811_144832280_05 KakaoTalk_20210811_144832280_06 KakaoTalk_20210811_144832280_07 KakaoTalk_20210811_144832280_08

---------------------- 8.18. test results 2 epochs

EnlightenGAN으로 조도개선 img 객체탐지 테스트

dst3_screenshot_02 08 2021 dst5_screenshot_02 08 2021 dst12_screenshot_11 08 2021 KakaoTalk_20210720_213727400

gray scale img 객체탐지 테스트

KakaoTalk_20210811_144832280 KakaoTalk_20210811_144832280_01 KakaoTalk_20210811_144832280_02 KakaoTalk_20210811_144832280_03 KakaoTalk_20210811_144832280_04 KakaoTalk_20210811_144832280_05 KakaoTalk_20210811_144832280_06 KakaoTalk_20210811_144832280_07 KakaoTalk_20210811_144832280_08

-------------------- 8.11. test results 1 epochs

YOLO data image test

000000016439_jpg rf 5799f1641ecbe657b77f74f6c11ab883
000000133645_jpg rf f349ed8b89e02746dfc2fc2f6a19d1ad
000000214192_jpg rf 2a14b7534dfe02879327b5a89f618cb2
000000276055_jpg rf d9804930ad7faf9c36f32d8dc2e602a5
000000329827_jpg rf a320319a40369375ac81ec62a8be5dc4
000000357978_jpg rf 91d4f5a74ed06116e2a0aea63d08da71
bus

EnlightenGAN으로 조도개선 img 객체탐지 테스트

dst3_screenshot_02 08 2021

gray scale img 객체탐지 테스트

KakaoTalk_20210811_144832280
KakaoTalk_20210811_144832280_01
KakaoTalk_20210811_144832280_02
KakaoTalk_20210811_144832280_03
KakaoTalk_20210811_144832280_05
KakaoTalk_20210811_144832280_06
KakaoTalk_20210811_144832280_08

모델 학습 결과(1 epochs)

P_curve

R_curve

PR_curve

F1_score

confusion_matrix

train_batch_0

train_batch0

train_batch_1

train_batch1

train_batch_2

train_batch2

validation_batch_labels_0

val_batch0_labels

validation_batch_predict_0

val_batch0_pred

validation_batch_labels_1

val_batch1_labels

validation_batch_predict_1

val_batch1_pred

validation_batch_labels_2

val_batch2_labels

validation_batch_predict_2

val_batch2_pred