diff --git a/README.md b/README.md index a743ddc34f8748cb29c92d65a63523ca01eed901..1e7687f1767e23bbf60535eecd5c215c39a97dcc 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