crowdcount.data.data_loader

class crowdcount.data.data_loader.ShanghaiTechDataset(mode='train', part='a', img_transform=None, gt_transform=None, both_transform=None, root='../crowd_count/data/datasets/shtu_dataset/')[source]

ShanghaiTech Dataset, Refer from “MCNN…” paper.

Parameters

  • mode (str, optional) – “train” | “test”, if “train”: load the train part data, if “test”: load the test part data(default: “train”).

  • part (str, optional) – “a” | “b”, if “a”: load the ShanghaiTech part A, if “b”: load the shanghaiTech part B(default: “a”).

  • img_transform (list of crowdcount.transform objects, optional) – transforms applied to image(default: None).

  • gt_transform (list of crowdcount.transform objects, optional) – transforms applied to ground truth(default: None).

  • both_transform (list of crowdcount.transform objects, optional) – transforms applied to both of image and ground truth(default:None).

  • root (str, optional) – the root directory of dataset(default: “../crowd_count/data/datasets/shtu_dataset/”).