boostcampaitech2 / object-detection-level2-cv-02

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

feat: Making pseudo label Training data with inferencing #44

Closed sejongjeong closed 3 years ago

sejongjeong commented 3 years ago

Pseudo Labeling

object-detection-level2-cv-02/template/mmdetection/tools -- inference_with_pseudo_labeling.py Inference을 진행하면서 동시에 Pseudo Label을 사용한 새로운 Train.json 파일 생성

argument

  1. config : config 파일 경로
  2. --new_train_path : peudo labeling 된 새로운 Json파일을 생성할 경로, default = "/opt/ml/detection/dataset/"
  3. --new_train_name : peudo labeling 된 새로운 json 파일을 생성할 이름, default = "pseudo_train.json"
  4. --original_train : Original train json 파일 경로, default = "/opt/ml/detection/dataset/train.json"
  5. --work_dir : work_dir 경로
  6. --epoch : 사용할 epoch, default = "latest" 7. --show_score_thr : 사용할 Threshold, default = 0.05 Train 데이터에 사용할 수 있는 수준의 박스 데이터를 도출하는 값이 필요하다.

이후 cocosplit을 사용하여 train, val set으로 split

close #13

Yiujin commented 3 years ago

수고하셨습니다!!!!!!

hojihun5516 commented 3 years ago

감사합니다