top-g-dev's picture
Duplicate from amirDev/crowd-counting-p2p
eeab6c4
raw
history blame contribute delete
No virus
221 Bytes
# build dataset according to given 'dataset_file'
def build_dataset(args):
if args.dataset_file == 'SHHA':
from crowd_datasets.SHHA.loading_data import loading_data
return loading_data
return None