Checkpoints for demo.py

Verg-Avesta / CounTR

CounTR: Transformer-based Generalised Visual Counting

https://verg-avesta.github.io/CounTR_Webpage/

MIT License

92 stars 9 forks source link

Checkpoints for demo.py #47

Closed InterstellarIntruder closed 3 months ago

InterstellarIntruder commented 3 months ago

Where can I download the checkpoints in the demo.py script?

checkpoint = torch.load('./output_allnew_dir/checkpoint-400.pth', map_location='cpu')
model_without_ddp.load_state_dict(checkpoint['model'], strict=False)
print("Resume checkpoint %s" % './output_allnew_dir/checkpoint-400.pth')