Added comment

This commit is contained in:
Zhengyi Chen 2024-03-02 23:40:18 +00:00
parent c88b938680
commit 5d77c1da4e

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
@ -130,4 +146,4 @@ def make_npydata():
if __name__ == "__main__":
# Download manually...
pre_dataset_sh() # XXX: preliminary
make_npydata()
make_npydata()