From 6aa3e061c40f9e589981e2ea6a150bb659bd4ac4 Mon Sep 17 00:00:00 2001 From: Jiwen Tang <rainbowend35@gmail.com> Date: Sat, 15 Oct 2022 21:14:02 +0800 Subject: [PATCH] update detection readme --- README.md | 13 ++++++++++++- 1 file changed, 12 insertions(+), 1 deletion(-) diff --git a/README.md b/README.md index a743ddc..1e7687f 100644 --- a/README.md +++ b/README.md @@ -44,7 +44,18 @@ Download the trained models at [Onedrive](https://mailstsinghuaeducn-my.sharepoi ``` ## Detection -TODO: tjw + +We adopt yolox as the detection method. We used stronger data augmentation and ranger optimizer. + +### Training + +Download the pretrained model at [Onedrive](https://mailstsinghuaeducn-my.sharepoint.com/personal/liuxy21_mails_tsinghua_edu_cn/_layouts/15/onedrive.aspx?ga=1&id=%2Fpersonal%2Fliuxy21%5Fmails%5Ftsinghua%5Fedu%5Fcn%2FDocuments%2Fbop%5Fchallenge%5F2022%2Fpretraied%5Fmodels%2Fyolox) (password: groupji) and put it in the folder `pretrained_models/yolox`. Then use the following command: + +`./det/yolox/tools/train_yolox.sh <config_path> <gpu_ids> (other args)` + +### Testing + +`./det/yolox/tools/test_yolox.sh <config_path> <gpu_ids> <ckpt_path> (other args)` ## Pose Estimation -- GitLab