search

JDAI-CV / CoTNet

This is an official implementation for "Contextual Transformer Networks for Visual Recognition".
https://arxiv.org/pdf/2107.12292.pdf
Other
514 stars 78 forks source link

CotNet Test #12

Closed amirhamidihd closed 3 years ago

amirhamidihd commented 3 years ago

How can I test the CotNet on a single image, there is no demo file on your git.

which file should be added to the detectron2 sources for running the demo test? where should I add the pre-trained model?

YehLi commented 3 years ago

You can refer to https://github.com/rwightman/pytorch-image-models for inference.

For detection and segmentation tasks, you can follow the guidance to test a single arbitrary image(https://detectron2.readthedocs.io/en/latest/tutorials/getting_started.html#inference-demo-with-pre-trained-models).