上面定义了一个Dataset对象CSDataSet，之后我们在train.py中定义DataLoader对象trainloader，并将CSDataSet作为参数传入。
trainloader = data.DataLoader(CSDataSet(args.data_dir, args.data_list, max_iters=args.num_steps*args.batch_size, crop_size=input_size, scale=args.random_scale, mirror=args.random_mirror, mean=IMG_MEAN), batch_size=args.batch_size, shuffle=True, num_workers=4, pin_memory=True)