221 Commits

Author SHA1 Message Date
KaiyangZhou
25c64fdd9a handle UnicodeDecodeError when using python3 2019-03-08 22:27:11 +00:00
KaiyangZhou
dc99054af5 compute mAP for cuhk03 old split with multi-gallery-shot 2019-03-07 09:42:06 +00:00
KaiyangZhou
7ff331f871 update mobilenetv2; add imagenet/reid weights 2019-03-05 10:56:39 +00:00
KaiyangZhou
3fe5cb118c polish function name 2019-02-27 09:57:48 +00:00
KaiyangZhou
83b9b38280 update ilids and prid 2019-02-27 09:31:12 +00:00
KaiyangZhou
4078f18aff update print msg 2019-02-20 22:28:30 +00:00
KaiyangZhou
5f971cc03a correct print 2019-02-20 22:22:46 +00:00
KaiyangZhou
c427e88658 rank={:.1%} 2019-02-20 22:20:36 +00:00
KaiyangZhou
d8cfa82594 add print to save_ckpt 2019-02-20 22:17:47 +00:00
KaiyangZhou
19e03d4097 add remove_module_from_keys option to save_ckpt 2019-02-20 22:13:26 +00:00
KaiyangZhou
d56e216068 bug fix 2019-02-20 21:57:09 +00:00
KaiyangZhou
74a2377ab7 add resume_from_ckpt 2019-02-20 21:51:49 +00:00
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