How to infer on an image?

[luckcodingdog]

Thank you for your great work!
Is there any script to infer on an image with your model?

[jozhang97]

Thanks for your interest!
I haven't written one myself but the nice people at HuggingFace (Niels) prepared one here
See this github issue #3

[luckcodingdog]

Thanks for your answer.
I have test the same script from Niels' github:
https://github.com/NielsRogge/Transformers-Tutorials/blob/master/DETA/Inference_with_DETA_(ResNet_%2B_Swin_backbones).ipynb
It run on cpu not gpu.
If someone wants to test on gpu, he should add lines like below:
device = "cuda" if torch.cuda.is_available() else "cpu"
model.to(device)
inputs = inputs.to(device)

[xinlin-xiao]

Thanks for your answer. I have test the same script from Niels' github: https://github.com/NielsRogge/Transformers-Tutorials/blob/master/DETA/Inference_with_DETA_(ResNet_%2B_Swin_backbones).ipynb It run on cpu not gpu. If someone wants to test on gpu, he should add lines like below: device = "cuda" if torch.cuda.is_available() else "cpu" model.to(device) inputs = inputs.to(device)

@luckcodingdog Hello， Where can I add these lines：
device = "cuda" if torch.cuda.is_available() else "cpu" model.to(device) inputs = inputs.to(device)