/DINO-ms

DINO implemented in mindspore

Primary LanguagePython

DINO-ms

DINO: DETR with Improved DeNoising Anchor Boxes for End-to-End Object Detection

network

Installation

The code requires python>=3.7 and mindspore>=2.1 and currently supports GPU platform. Please follow the instructions here to install mindspore dependencies.

Clone the repository locally and install with

git clone https://github.com/Mark-ZhouWX/models.git
pip install -r requirements.txt

Train

First put your dataset under {project_root}/datasets (currently only COCO is supported), and download pretrained model from here, then run:

python train.py

for single card training, or

mpirun --allow-run-as-root -n 8 python train.py

for multi-card training

Inference

For inference, please run:

python eval.py --eval_model_path /path/to/model