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 |
|
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 |