c388c248801797454eba162c96306103c2c632e0,train_vidreid_xent_htri.py,,main,#,110

Before Change

    print("Initializing dataset {}".format(args.dataset))
    dataset = data_manager.init_vidreid_dataset(root=args.root, name=args.dataset)

    transform_train = T.Compose([
        T.Random2DTranslation(args.height, args.width),
        T.RandomHorizontalFlip(),
        T.ToTensor(),
        T.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]),
    ])

    transform_test = T.Compose([
        T.Resize((args.height, args.width)),
        T.ToTensor(),

After Change

    print("Initializing dataset {}".format(args.dataset))
    dataset = data_manager.init_vidreid_dataset(root=args.root, name=args.dataset)

    transform_train = build_transforms(args.height, args.width, is_train=True)
    transform_test = build_transforms(args.height, args.width, is_train=False)

    pin_memory = True if use_gpu else False

Instances