Are you sure you want to delete this task? Once this task is deleted, it cannot be recovered.
Jia-Xin Zhuang dcc67f7d1b | 1 year ago | |
---|---|---|
.. | ||
configs | 1 year ago | |
detr | 1 year ago | |
README.md | 1 year ago | |
converter.py | 1 year ago | |
train_net.py | 1 year ago |
We provide a Detectron2 wrapper for DETR, thus providing a way to better integrate it in the existing detection ecosystem. It can be used for example to easily leverage datasets or backbones provided in Detectron2.
This wrapper currently supports only box detection, and is intended to be as close as possible to the original implementation, and we checked that it indeed match the results. Some notable facts and caveats:
To install Detectron2, please follow the official installation instructions.
For convenience, we provide a conversion script to convert models trained by the main DETR training loop into the format of this wrapper. To download and convert the main Resnet50 model, simply do:
python converter.py --source_model https://dl.fbaipublicfiles.com/detr/detr-r50-e632da11.pth --output_model converted_model.pth
You can then evaluate it using:
python train_net.py --eval-only --config configs/detr_256_6_6_torchvision.yaml MODEL.WEIGHTS "converted_model.pth"
To train DETR on a single node with 8 gpus, simply use:
python train_net.py --config configs/detr_256_6_6_torchvision.yaml --num-gpus 8
To fine-tune DETR for instance segmentation on a single node with 8 gpus, simply use:
python train_net.py --config configs/detr_segm_256_6_6_torchvision.yaml --num-gpus 8 MODEL.DETR.FROZEN_WEIGHTS <model_path>
LungNodules detection framework based on DeTR
Jupyter Notebook Text CSV Python
Dear OpenI User
Thank you for your continuous support to the Openl Qizhi Community AI Collaboration Platform. In order to protect your usage rights and ensure network security, we updated the Openl Qizhi Community AI Collaboration Platform Usage Agreement in January 2024. The updated agreement specifies that users are prohibited from using intranet penetration tools. After you click "Agree and continue", you can continue to use our services. Thank you for your cooperation and understanding.
For more agreement content, please refer to the《Openl Qizhi Community AI Collaboration Platform Usage Agreement》