quyang-hub / vehicle-density-estimation

OSTNet: Overlapping Splitting Transformer Network with Integrated Density Loss for Vehicle Density Estimation

1.Environment Setup

Python library dependencies.

-f https://download.pytorch.org/whl/cu113/torch_stable.html
torch==1.10.0+cu113
torchvision==0.11.1+cu113
torchaudio==0.10.0+cu113 
numpy>=1.16.5
scipy>=1.3.0
opencv-python
gdown
pillow
gradio
timm==0.4.12
wandb
matplotlib

Train

• train.py

python train.py

If you want to replace the dataset, you can use the variable dataset = 'dataset name' in the train.py file. There are also a number of parameters that can be modified to suit your needs, such as learning rate, epochs, etc.

Test

Download our trained model, Trancos.

If you want to use our model quickly, you can use the test.py file directly and modify the path to the model in the file to get our results quickly.

• test.py

python test.py

If you want to replace the dataset, you can use the variable dataset = 'dataset name' in the test.py file.