관리 메뉴

ComputerVision Jack

Public Dataset (Wider Face) 본문

Computer Vision/Public Data

Public Dataset (Wider Face)

JackYoon 2021. 5. 3. 21:34
반응형

shuoyang1213.me/WIDERFACE/

 

WIDER FACE: A Face Detection Benchmark

News 2017-03-31 The new version of evaluation code and validation resultsis released. 2017-03-31 Add text version ground truth and fix rounding problem of bounding box annotations. 2016-08-19 Two new algorithms are added into leader-board. 2016-04-17 The f

shuoyang1213.me

오늘 소개하는 Dataset은 Wider Face Dataset입니다.

Detection Task를 진행할 때, 가장 많이 시도하는 작업은 Face Detection입니다. 하지만 Face Dataset 수집은 쉽지 않습니다. 저작권의 문제도 발생하기 때문입니다. 따라서 Public Dataset으로 유명한 Wider Face Dataset을 소개합니다.

 

해당 site에 들어가시면 Dataset을 다운로드할 수 있습니다.

그러면 WIDER_train, WIDER_val, WIDER_test 3개의 이미지 파일과 wider_face_split의 annotation이 담긴 파일을 다운로드할 수 있습니다.

 

wider_face_train_bbx_gt.txt 파일을 참고하시면 해당 image 파일에 대한 annotation이 매칭되어 있습니다.

annotation : x1, y1, w, h, blur, expression, illumination, invalid, occlusion, pose 

 

저는 Face Detection을 수행할 때, 이 중에서 x1, y1, w, h만 사용하였고, 해당 이미지 파일에 txt 파일을 따로 생성하여 label값을 넣어주었습니다.

 

[Annotation 파일 처리]

Annotation 처리

해당 코드는 Annotation 파일을 읽어서 JPG 파일 밑에 TXT 파일을 생성하는 코드입니다.

 

Annotation 수정

저는 Face에 대한 Bounding box로 접근하기 때문에 cx, cy, width, height로 해당 Annotation 파일을 읽어 변환해주었습니다.

반응형

'Computer Vision > Public Data' 카테고리의 다른 글

Public Dataset (COCO)  (0) 2021.06.14
Public Dataset (VGG Face)  (1) 2021.05.06
Comments