update1
@ -0,0 +1,58 @@
|
||||
# 지능화 캡스톤 프로젝트 #2 - YOLOv5 안전모 검출
|
||||
|
||||
-----
|
||||
|
||||
|
||||
[PINBlog Gitea Repository](https://gitea.pinblog.codes/CBNU/19_YOLOv5)
|
||||
* 제출했던 프로젝트 코드가 날라가서 일부만 존재함;;
|
||||
|
||||
-----
|
||||
|
||||
### YOLOv5 GitHub
|
||||
[YOLOv5](https://github.com/ultralytics/yolov5)
|
||||
|
||||
|
||||
# 프로젝트 목표
|
||||
* YOLOv5 모델을 학습하여 영상의 안전모와 마스크를 검출
|
||||
|
||||
# 구현 방법
|
||||
* 일반적인 1개의 모델로 여러 클래스를 학습하지 않고, Detect Model로 머리를 검출하여 ROI 취득 후,
|
||||
해당 ROI를 Crop하여 Classification Model로 안전모, 마스크, 미착용 등을 분류한다.
|
||||
|
||||
# 데이터셋
|
||||
* YouTube, Google, AIHub, Kaggle에서 안전모 관련 데이터셋 확보
|
||||
* Roboflow에서 데이터셋 라벨링 진행
|
||||
[Roboflow](https://roboflow.com/)
|
||||
|
||||
# 클래스
|
||||
### Detect Model
|
||||
* 1개의 뚝배기 클래스
|
||||
* 0: dduk
|
||||
|
||||
### Classification Model
|
||||
* 5개의 클래스
|
||||
* 0: head
|
||||
* 1: helmet
|
||||
* 2: face
|
||||
* 3: mask
|
||||
* 4: helmet & mask
|
||||
|
||||
# 프로젝트 코드 (Validation)
|
||||
|
||||
<details>
|
||||
<summary></summary>
|
||||
<div markdown="1">
|
||||
|
||||
|
||||
``` planetext
|
||||
|
||||
|
||||
```
|
||||
|
||||
</div>
|
||||
</details>
|
||||
|
||||
|
||||
### 참고[¶]()
|
||||
|
||||
- 지능화캡스톤 과목, 김현용 교수
|
Before Width: | Height: | Size: 90 KiB After Width: | Height: | Size: 90 KiB |
Before Width: | Height: | Size: 90 KiB After Width: | Height: | Size: 90 KiB |
Before Width: | Height: | Size: 88 KiB After Width: | Height: | Size: 88 KiB |
Before Width: | Height: | Size: 91 KiB After Width: | Height: | Size: 91 KiB |
Before Width: | Height: | Size: 80 KiB After Width: | Height: | Size: 80 KiB |
Before Width: | Height: | Size: 125 KiB After Width: | Height: | Size: 125 KiB |
Before Width: | Height: | Size: 264 KiB After Width: | Height: | Size: 264 KiB |
Before Width: | Height: | Size: 258 KiB After Width: | Height: | Size: 258 KiB |
Before Width: | Height: | Size: 636 KiB After Width: | Height: | Size: 636 KiB |
Before Width: | Height: | Size: 607 KiB After Width: | Height: | Size: 607 KiB |
Before Width: | Height: | Size: 619 KiB After Width: | Height: | Size: 619 KiB |
Before Width: | Height: | Size: 518 KiB After Width: | Height: | Size: 518 KiB |
Before Width: | Height: | Size: 532 KiB After Width: | Height: | Size: 532 KiB |
Before Width: | Height: | Size: 642 KiB After Width: | Height: | Size: 642 KiB |
Before Width: | Height: | Size: 655 KiB After Width: | Height: | Size: 655 KiB |
Before Width: | Height: | Size: 530 KiB After Width: | Height: | Size: 530 KiB |
Before Width: | Height: | Size: 572 KiB After Width: | Height: | Size: 572 KiB |