Skip to content

Latest commit

聽

History

History
16 lines (13 loc) 路 828 Bytes

File metadata and controls

16 lines (13 loc) 路 828 Bytes

Setup the Environment

Please follow the instruction at Sense-X/Co-DETR to setup the environment.

Download Pretrained Checkpoints

Please use google drive link to download co_deformable_detr_swin_large_900q_3x_coco.pth checkpoints.

Command to Run

python launch_codetr_multi_gpu_inference.py --image_dir_path <path to the directory containing images> --output_dir_path <base path to store the predictions> --ckpt_path <Path to the checkpoints - co_deformable_detr_swin_large_900q_3x_coco.pth> --gpu_ids <comma separated gpu ids to run inference on multiple gpus>

For help, run

python launch_codetr_multi_gpu_inference.py -h
python infer.py -h