318 Commits

Author SHA1 Message Date
liaoxingyu
a5550d7725 fix balanced sampler bug
Summary: fix typing error in balanced sampler with missing `len` in ret.
2020-08-19 16:28:10 +08:00
liaoxingyu
fd06a2819a delete sh file 2020-08-14 19:07:46 +08:00
liaoxingyu
f29bc025cc update readme for distill 2020-08-14 19:06:36 +08:00
Xingyu Liao
c01b859205
support distillation
Merge distillation code from @wangguanan 

Reviewed by: l1aoxingyu
2020-08-14 17:31:04 +08:00
liaoxingyu
9c667d1a0f add pretrain_path support in backbone 2020-08-14 14:00:26 +08:00
liaoxingyu
e0cf8ac56e fix sampler problem #217
Fix problem about periodic long waiting time when number of images is large.

In the old version, it will prepare the whole epoch indices when finishing one epoch.
Now it changes to prepare the current batch indices.
2020-08-14 13:57:22 +08:00
liaoxingyu
ae7c9288cf support faiss retrieval and cython roc evaluation 2020-08-12 16:27:57 +08:00
Xingyu Liao
f74cebcd88
fix rank_cylib bug about valid query index
fix bugs of rank_cylib

Reviewed by: l1aoxingyu
2020-08-10 19:28:05 +08:00
liaoxingyu
a198567f27 update ibn-net pretrain path 2020-08-10 14:51:35 +08:00
liaoxingyu
db6b42da4f update resnest url 2020-08-10 14:18:00 +08:00
wangguanan
dc9d1e39d0 add msmt17 results 2020-08-10 11:04:28 +08:00
wangguanan
bbbcdb35ea add msmt17 results 2020-08-10 10:54:42 +08:00
wangguanan
52cfd1cc3f add msmt17 results 2020-08-10 10:49:20 +08:00
wangguanan
49bd511750 add msmt17 results 2020-08-10 10:35:00 +08:00
wangguanan
7ddb5b6f59 add msmt17 results 2020-08-10 10:31:41 +08:00
liaoxingyu
57ebf75f83 fix bug about changing cfg values 2020-08-10 10:29:24 +08:00
wangguanan
531ecf3782 fix config bug 2020-08-10 10:10:47 +08:00
liaoxingyu
d516d59fb5 fix autoaug total_iter bug #219
max_epoch does not convert to max_iter when needing to build dataloader first
add extra convert in `build_reid_train_loader`
2020-08-09 19:43:01 +08:00
wangguanan
0aa1425167 fix 34-kd-18 bugs 2020-08-07 11:30:43 +08:00
wangguanan
5856da3d09 fix 34-kd-18 bugs 2020-08-07 11:23:07 +08:00
wangguanan
69b7a087e7 update market1501 results 2020-08-07 11:21:03 +08:00
wangguanan
1d6778333c update market1501 results 2020-08-07 11:19:02 +08:00
wangguanan
d6d1b84096 update market1501 results 2020-08-07 11:18:13 +08:00
wangguanan
d6b7bf7c50 update market1501 results 2020-08-07 10:35:26 +08:00
wangguanan
1b0f5fdd85 add distill-reid 2020-08-06 17:54:30 +08:00
wangguanan
7a58effeb5 add distill-reid 2020-08-06 17:49:27 +08:00
wangguanan
5adb7173cb add distill-reid 2020-08-06 17:44:05 +08:00
wangguanan
3426bec5b0 add distill-reid 2020-08-06 17:42:39 +08:00
wangguanan
d4b3eff907 add distill-reid 2020-08-06 17:40:28 +08:00
wangguanan
0fa4f3c996 add distill-reid 2020-08-06 17:39:34 +08:00
wangguanan
6bc0913f83 add distill-reid 2020-08-06 17:39:06 +08:00
wangguanan
c00967b6c4 add distill-reid 2020-08-06 17:28:39 +08:00
wangguanan
5962289801 add distill-reid 2020-08-06 17:26:21 +08:00
wangguanan
5bf228ae5d add distill-reid 2020-08-06 16:24:37 +08:00
liaoxingyu
3549685c4c update url 2020-08-04 16:00:12 +08:00
liaoxingyu
d1c20cbe50 fix pre-train model bugs
fix bugs locks when downloading pre-train model
2020-08-04 15:56:36 +08:00
liaoxingyu
e1aaeb358b rm redundant export 2020-07-31 16:37:16 +08:00
liaoxingyu
35794621cc remove addmm warning 2020-07-31 16:32:10 +08:00
liaoxingyu
5e23ab756d update model url 2020-07-31 15:07:10 +08:00
liaoxingyu
977aa4a357 update model url 2020-07-31 14:59:37 +08:00
liaoxingyu
2430b8ed75 pretrain model bugfix
Fix pretrain model download bugs and testing bugs in multiprocess
2020-07-31 10:42:38 +08:00
liaoxingyu
1ed73fb238 add pretrain model url
close #199 #191
2020-07-30 20:19:42 +08:00
liaoxingyu
65169b40bd support ddp testing 2020-07-30 20:15:28 +08:00
liaoxingyu
64861ea6c8 update config file v0.1.1 2020-07-29 19:48:49 +08:00
liaoxingyu
a8c7e07df6 update deploy docs 2020-07-29 17:47:35 +08:00
liaoxingyu
16655448c2 onnx/trt support
Summary: change model pretrain mode and support onnx/TensorRT export
2020-07-29 17:43:39 +08:00
liaoxingyu
ee634df290 rm _all_ in resnet 2020-07-17 19:40:25 +08:00
liaoxingyu
5ad22d5d36 update regnet init 2020-07-17 19:35:40 +08:00
liaoxingyu
f9539be683 add regnet config file 2020-07-17 19:32:36 +08:00
liaoxingyu
e269caf8ab update dsr head 2020-07-17 19:14:20 +08:00