144 Commits

Author SHA1 Message Date
kaiyangzhou
5cc8043c4a verbose=False after __add__ 2019-03-25 20:03:21 +00:00
KaiyangZhou
866744914b 0.7.0 -> 0.7.1 2019-03-25 18:49:20 +00:00
KaiyangZhou
235348f672 bug fix in __add__ 2019-03-25 18:43:19 +00:00
KaiyangZhou
1e7bbc3668 add \n 2019-03-25 16:27:39 +00:00
KaiyangZhou
8e5265099d change weight path 2019-03-25 09:29:06 +00:00
KaiyangZhou
e5b8c464ca update docstrings 2019-03-24 23:33:54 +00:00
KaiyangZhou
4ca182dbb8 add model public keys 2019-03-24 15:52:39 +00:00
KaiyangZhou
a0e54ede32 add scripts 2019-03-24 11:06:12 +00:00
KaiyangZhou
d0fe37c558 finalize docs 2019-03-23 23:09:39 +00:00
KaiyangZhou
aa98ef5a7b add docstrings to losses, metrics, models, utils, optim 2019-03-22 00:14:41 +00:00
KaiyangZhou
cbbde86eef add docs to torchreid.data 2019-03-21 17:28:14 +00:00
KaiyangZhou
75dd835e42 change batch_size in video 2019-03-21 14:12:50 +00:00
KaiyangZhou
77c47b4e53 add data module 2019-03-21 13:00:07 +00:00
KaiyangZhou
46a0d5c27d update dataset and docs 2019-03-21 12:53:21 +00:00
KaiyangZhou
37f92b7cc3 create engine SDK 2019-03-19 17:26:08 +00:00
KaiyangZhou
03d0674dcb comply woth PEP8 2019-03-15 23:17:38 +00:00
KaiyangZhou
983c9e51e5 add --combineall 2019-03-15 17:45:47 +00:00
KaiyangZhou
402e054110 rearrange order of functions 2019-03-15 16:20:47 +00:00
KaiyangZhou
3cbeab42c2 add init_attributes to BaseDataset 2019-03-15 16:11:25 +00:00
KaiyangZhou
e2b1001165 polish code 2019-03-15 14:49:18 +00:00
KaiyangZhou
850da2bbed update path to pretrained weights of resnext 2019-03-12 16:46:49 +00:00
KaiyangZhou
6f3b057c1b add space between class and __init__ (trivial) 2019-03-09 12:53:23 +00:00
KaiyangZhou
5cd0776eb5 rm center loss & ring loss 2019-03-09 12:45:33 +00:00
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