search

boostcampaitech2 / object-detection-level2-cv-02

object-detection-level2-cv-02 created by GitHub Classroom
2 stars 4 forks source link

feat: Group stratified k-fold #39

Closed Yiujin closed 3 years ago

Yiujin commented 3 years ago

mmdetection/tools make_fold_annotation.py : make k annontation files(train, val) train_cv.py : train with k fold inference_cv.py : inference with k times and ensemble all fold result

  1. make_fold_annotation.py 돌려서 annotation file 만들기 --annotations 인자는 부캠에서 준 원래 train.json 파일 위치 지정해주세요

  2. train_cv.py 로 학습 돌리기 wandb 는 각 fold마다 하나씩 총 k=3개가 만들어져요

  3. inference.py 로 각 폴드별 best pth로 csv 파일 만든 후 앙상블, 앙상블은 nms 입니다.

resolve #7

JODONG2 commented 3 years ago

고생했슴다 잘쓰겠씀다 !

binlee52 commented 3 years ago

감사합니다. 서버를 더 열심히 괴롭히겠습니다.