$ docker compose exec dethub python tools/image_demo.py configs/projects/coco/demo/000000000025.jpg configs/projects/coco/yolox/yolox_s_swin_s_coco.py --weights https://github.com/okotaku/dethub-weights/releases/download/v0.1.1cocoyolox/yolox_s_swin_s-19d4228f.pth --out-dir configs/projects/coco/demo/result
- Download data from official page
wget http://images.cocodataset.org/zips/train2017.zip
wget http://images.cocodataset.org/zips/val2017.zip
wget http://images.cocodataset.org/annotations/annotations_trainval2017.zip
- Unzip the files as follows
data/coco
├── annotations
├── train2017
└── val2017
Set env variables
$ export DATA_DIR=/path/to/data
Start a docker container
$ docker compose up -d dethub
Run train
# single gpu
$ docker compose exec dethub mim train mmdet configs/projects/coco/rpn/atss_rpn_r50_fpn.py
# multi gpus
$ docker compose exec dethub mim train mmdet configs/projects/coco/rpn/atss_rpn_r50_fpn.py --gpus 2 --launcher pytorch