zc/mlp-project
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Added comment

Browse source
This commit is contained in:
Zhengyi Chen 2024-03-02 23:40:18 +00:00
parent c88b938680
commit 5d77c1da4e
1 changed files with 17 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

16
make_dataset.py
View file

@ -1,3 +1,19 @@
"""
The TransCrowd paper lists ShanghaiTech dataset as from here:
https://drive.google.com/file/d/1CkYppr_IqR1s6wi53l2gKoGqm7LkJ-Lc/view
Alternatively, you could prob. download from here:
https://www.kaggle.com/datasets/tthien/shanghaitech?resource=download
After downloading, execute:
$ unzip <downloaded-zip-file> -d <repo-dir>/synchronous/dataset/
To unzip the dataset correctly prior to running this script.
"""
import os
import glob
import random