159 Commits

Author	SHA1	Message	Date
KaiyangZhou	ae54ca0f6b	mv save_ckpt to torchtools	2019-02-20 21:51:24 +00:00
KaiyangZhou	26a9400609	change double quotes to single quote in print	2019-02-20 21:50:47 +00:00
kaiyangzhou	356a677cd3	mv basic args to base-manager	2019-02-19 15:48:10 +00:00
kaiyangzhou	1cdb312007	add --color-jitter & --color-aug	2019-02-19 15:44:56 +00:00
kaiyangzhou	859b03cf19	set pretrained=True instead of pretrained="imagenet"	2019-02-19 11:10:45 +00:00
KaiyangZhou	2a5a951666	wrap lr_scheduler	2019-02-08 11:44:12 +00:00
KaiyangZhou	3d67488edd	support staged_lr	2019-02-03 14:04:10 +00:00
KaiyangZhou	cf5b9a3ba5	update comments	2019-02-03 14:03:48 +00:00
KaiyangZhou	bac2ce5662	add random erasing	2019-02-03 14:03:06 +00:00
KaiyangZhou	9c159569fd	update author info	2019-02-03 14:01:26 +00:00
KaiyangZhou	043c820d01	ignore "module." when saving/loading weights	2019-02-02 14:20:56 +00:00
KaiyangZhou	760d68ebd1	bug fix	2019-02-01 22:43:47 +00:00
KaiyangZhou	6891a6424a	mv model.cuda() before init_optim()	2019-02-01 22:30:41 +00:00
KaiyangZhou	2cb1031942	add load_pretrained_weights() to torchtools	2019-02-01 13:10:23 +00:00
KaiyangZhou	800df43fee	check if output is list	2019-02-01 00:50:28 +00:00
KaiyangZhou	564f5a26ab	update print	2019-01-30 22:41:47 +00:00
KaiyangZhou	b2c06364a3	unify sampler	2019-01-28 18:50:09 +00:00
KaiyangZhou	daf184608c	add self.	2019-01-28 17:29:22 +00:00
KaiyangZhou	f045778602	add accuracy	2019-01-27 23:15:38 +00:00
KaiyangZhou	25aecab2da	bug fix	2019-01-24 22:12:02 +00:00
KaiyangZhou	825fce92eb	support msmt_v2	2019-01-22 18:18:59 +00:00
kaiyangzhou	ef2bacbc00	add market1501-500k	2019-01-22 16:07:43 +00:00
KaiyangZhou	4b6057b923	wrap set_random_seed	2019-01-09 21:30:46 +00:00
KaiyangZhou	10b074e9b7	add os.path.expanduser(root) to dataset files	2019-01-09 13:43:39 +00:00
KaiyangZhou	074a10e24e	bug fix	2018-12-27 10:25:09 +00:00
KaiyangZhou	f8a31da4c6	replace self.pin_memory with self.use_gpu	2018-12-26 11:10:34 +00:00
KaiyangZhou	c05f96e31b	add link to shufflenet's imagenet weights	2018-11-26 22:07:20 +00:00
KaiyangZhou	47b883d0d6	bug fix: rm resnext101_32x4d	2018-11-26 22:06:33 +00:00
KaiyangZhou	534dfa6cce	update code: add self to image_training	2018-11-24 16:48:07 +00:00
KaiyangZhou	9c373ff6e6	bug fix: add one output for unpacking trainloader	2018-11-24 16:25:10 +00:00
KaiyangZhou	affeaa847d	add resnext50_32x4d's imagenet weights	2018-11-23 10:31:38 +00:00
KaiyangZhou	d7755cfd16	add resnext50_32x4d_fc512	2018-11-22 12:41:22 +00:00
KaiyangZhou	b065f130d0	rename tracklet_info to tracklet_stats in datasets/base.py	2018-11-21 09:59:43 +00:00
KaiyangZhou	e9a2b7c3d3	add overhead sign	2018-11-12 11:45:41 +00:00
KaiyangZhou	d8498ea3dd	optimize cython code	2018-11-12 11:44:20 +00:00
KaiyangZhou	8ca8be28c2	update test_cython.py: add cuhk03's comparison	2018-11-11 23:02:20 +00:00
KaiyangZhou	8631603539	rm *.c files for cython	2018-11-11 17:43:28 +00:00
KaiyangZhou	7fe189f00f	add torchreid/eval_cylib/test_cython.py	2018-11-10 22:01:25 +00:00
KaiyangZhou	75279f0728	add evaluate_py	2018-11-10 21:25:40 +00:00
KaiyangZhou	41f1f6c6ac	fix bug: add return	2018-11-10 21:16:31 +00:00
KaiyangZhou	d885f11ce7	update cython code: add cuhk03-metric	2018-11-10 21:09:13 +00:00
KaiyangZhou	1d0851e8c4	fix bug: add self.num_instances to ImageDataManger	2018-11-10 11:59:01 +00:00
KaiyangZhou	8c53007138	return (imgs, pids, camids, img_paths) for image dataloaders	2018-11-10 11:54:06 +00:00
KaiyangZhou	db157c1345	update help comments for data_source	2018-11-10 11:53:25 +00:00
KaiyangZhou	0545358b6f	mv resnet50mid to reid-specific model list	2018-11-09 16:17:29 +00:00
KaiyangZhou	54b92ee617	check if isinstance(model, nn.DataParallel)	2018-11-08 23:54:26 +00:00
KaiyangZhou	205df55987	add fixbase functions (debug mode)	2018-11-08 23:04:50 +00:00
KaiyangZhou	cac6d72a6d	minor change: data_source (Dataset) -> data_source (list)	2018-11-08 22:34:24 +00:00
KaiyangZhou	9980b5378b	add support for htri training	2018-11-08 22:02:53 +00:00
KaiyangZhou	01fe76cf27	sample -> sample_method	2018-11-08 21:40:44 +00:00