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