search

boostcampaitech6 / level1-imageclassification-cv-10

level1-imageclassification-cv-10 created by GitHub Classroom
1 stars 2 forks source link

feat : face_detection by yolo #29

Closed classaen7 closed 8 months ago

classaen7 commented 8 months ago

feat

data/preprocess/detection.py detection에 대한 커맨드가 주어지면 훈련 데이터셋에서 Dataset을 생성할 때 detection을 수행한다.

커맨드 예시 python train.py --detection yolo

rembg에 대한 수정 필요