file examples

[ly-jy]

Hello author, how do you get 'market1501_init_-1.0.pickle','duke_init_-1.0.pickle' in file examples? Plase help me! Please! Thank you very much!

[Yu-Wu]

One-Example-Person-ReID/my_reid/eug.py

Lines 256 to 300 in f792dde

	label_dataset = []
	unlabel_dataset = []
	if init_str == "one-shot":
	# dataset indexed by [pid, cam]
	dataset_in_pid_cam = [[[] for _ in range(dataset.num_cams)] for _ in range(dataset.num_train_ids) ]
	for index, (images, pid, camid, videoid) in enumerate(dataset.train):
	dataset_in_pid_cam[pid][camid].append([images, pid, camid, videoid])
	# generate the labeled dataset by randomly selecting a tracklet from the first camera for each identity
	for pid, cams_data in enumerate(dataset_in_pid_cam):
	for camid, videos in enumerate(cams_data):
	if len(videos) != 0:
	selected_video = random.choice(videos)
	break
	label_dataset.append(selected_video)
	assert len(label_dataset) == dataset.num_train_ids
	labeled_videoIDs =[vid for _, (_,_,_, vid) in enumerate(label_dataset)]
	else:
	# dataset indexed by [pid]
	dataset_in_pid = [ [] for _ in range(dataset.num_train_ids) ]
	for index, (images, pid, camid, videoid) in enumerate(dataset.train):
	dataset_in_pid[pid].append([images, pid, camid, videoid])
	for pid, pid_data in enumerate(dataset_in_pid):
	k = int(np.ceil(len(pid_data) * init)) # random sample ratio
	selected_video = random.sample(pid_data, k)
	label_dataset.extend(selected_video)
	labeled_videoIDs =[vid for _, (_,_,_, vid) in enumerate(label_dataset)]
	# generate unlabeled set
	for (imgs, pid, camid, videoid) in dataset.train:
	if videoid not in labeled_videoIDs:
	unlabel_dataset.append([imgs, pid, camid, videoid])
	with open(load_path, "wb") as fp:
	pickle.dump({"label set":label_dataset, "unlabel set":unlabel_dataset}, fp)
	print(" labeled | N/A | {:8d}".format(len(label_dataset)))
	print(" unlabeled | N/A | {:8d}".format(len(unlabel_dataset)))
	print("\nCreate new {} split and save it to {}".format(init_str, load_path))
	return label_dataset, unlabel_dataset

Please see the code here.