search

IDEA-Research / DINO

[ICLR 2023] Official implementation of the paper "DINO: DETR with Improved DeNoising Anchor Boxes for End-to-End Object Detection"
Apache License 2.0
2.1k stars 230 forks source link

How to detect image on our checkpoint.pth #173

Open chos1npc opened 1 year ago

chos1npc commented 1 year ago

I want to detect our test set, but I can't find the detect.py or other code can detect image on our image, how should I do for that, Thx

SlongLiu commented 1 year ago

please refer to https://github.com/IDEA-Research/DINO/blob/main/inference_and_visualization.ipynb for inference

YejinHwang909 commented 1 year ago

@SlongLiu You don't have a detect.py, do you? If we want to save the results of the inference, we can refer to this ipynb file and implement on our own, right?