search

Sense-X / Co-DETR

[ICCV 2023] DETRs with Collaborative Hybrid Assignments Training
MIT License
950 stars 100 forks source link

start training #133

Closed prsbsvrn closed 4 months ago

prsbsvrn commented 4 months ago

Hello Thank you for the codes! I installed mmcv and I tried to implement co-deformable-detr but couldn't. Can you please provide an orderly structure for implementation co-deformable-detr?

prsbsvrn commented 4 months ago

Hi, I find the answer:

git clone https://github.com/Sense-X/Co-DETR.git cd Co-DETR/docker docker build -t codetr-image . docker run --shm-size=256m --gpus all -it --name codetr-container -v {directory of coco dataset}:/Co-DETR/data/coco -v {directory of co-detr github project}:/Co-DETR codetr-image

export SKLEARN_ALLOW_DEPRECATED_SKLEARN_PACKAGE_INSTALL=True pip install -r requirements.txt

Write "#!/bin/bash" in the first line of tools/dist_train.sh export PYTHONPATH="/Co-DETR:$PYTHONPATH" for 1 gpu: bash tools/dist_train.sh projects/configs/co_deformable_detr/co_deformable_detr_r50_1x_coco.py 1 path_to_exp