This is an attemp to replicate and amke the code form the CounTR project to work [GitHub Repo].
Official PyTorch implementation for CounTR. Details can be found in the paper. [Paper] [Project page]
You can download the weight from the CounTR project page, and the FSC147 dataset from its original project(link can also be found in the CounTR project page).
- Create Object counter for the density map to generate the total object detected
- Test pretraining the model.