Commit Graph

101 Commits (7076f1a750d58b427bd482bbc6a1b01e183d369d)

Author SHA1 Message Date
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
KaiyangZhou e17bbcddf3 source/target -> source_names/target_names 2018-11-08 21:40:18 +00:00
KaiyangZhou c32cf80a9f add RankLogger 2018-11-08 21:38:22 +00:00
KaiyangZhou fdc451c0e5 add linear decay 2018-11-07 22:02:23 +00:00
KaiyangZhou 2d32f5b28a add warnings 2018-11-07 21:55:04 +00:00
KaiyangZhou e3ae601ffc mv KeyError to ValueError 2018-11-07 21:54:46 +00:00
KaiyangZhou 7b47e645e9 add property 2018-11-07 21:46:23 +00:00
KaiyangZhou 1f1f01669b change dataset to (query, gallery) 2018-11-07 21:45:48 +00:00
KaiyangZhou 3f9e20a55e change access to dataloaders in dm by class method 2018-11-07 20:48:21 +00:00