Passion-ate🔥

강재현	김민준	박상현	서광채	오하은	이승우
	Blog	Blog	Notion		Notion

Overview

Project Structure

---

Result

• private LB : 0.666 (6등)

How to Use

Installation

• pip install -r requirements.txt

Model

Swin

• Train
  • python mmdetection/tools/train.py {CONFIG_PATH}
• Inference
  • python mmdetection/tools/test.py ${config} ${model} --out mmdetection/work_dirs/latest.pkl
• Evaluation
  • python mmdetection/make_submission.py --pkl mmdetection/work_dirs/latest.pkl --csv mmdetection/work_dirs/result.csv

@article{liu2021Swin,
  title={Swin Transformer: Hierarchical Vision Transformer using Shifted Windows},
  author={Liu, Ze and Lin, Yutong and Cao, Yue and Hu, Han and Wei, Yixuan and Zhang, Zheng and Lin, Stephen and Guo, Baining},
  journal={arXiv preprint arXiv:2103.14030},
  year={2021}
}

Yolor

• Train
  • python yolor/train.py --data yolor/trash_data/coco.yaml --cfg yolor/models/yolor-d6.yaml --weights 'yolor/yolor-d6.pt' --device 0 --name yolor-d6 --hyp trash_data/hyp.yaml
• Inference
  • python yolor/detect.py --source yolor/trash_data/images/test --weights yolor/runs/train/yolor-d6/weights/best.pt --conf 0.001 --iou 0.5 --img-size 1024 --device 0 --save-txt --save-conf --project yolor/runs/yolor-d6
• Evaluation
  • python yolor/yolor_make_submission.py

@article{wang2021you,
  title={You Only Learn One Representation: Unified Network for Multiple Tasks},
  author={Wang, Chien-Yao and Yeh, I-Hau and Liao, Hong-Yuan Mark},
  journal={arXiv preprint arXiv:2105.04206},
  year={2021}
}

PR curve Analysis

• json
  • python tools/test.py ${CONFIG} ${CHECKPOINT_DIR} --format-only --options "jsonfile_prefix=./${FILE NAME}"
• coco analysis
  • python tools/analysis_tools/coco_error_analysis.py ${RESULT} ${OUT_DIR} --ann ${ANN}