Commit Graph

482 Commits (bb6ddbf8b15520bae9090827f0ab3dc8cf3b1337)
 

Author SHA1 Message Date
liaoxingyu f06c8b2ed9 fix bug: make CJ works #239
fix #239
2020-08-25 11:31:49 +08:00
Xingyu Liao a2fe1f5271
fix typro
fix typro in README
2020-08-24 23:28:19 +08:00
liaoxingyu 9fb3285d5c rm extra sh file 2020-08-21 18:00:36 +08:00
liaoxingyu 198592a202 Merge branch 'master' of github.com:L1aoXingyu/fast-reid 2020-08-21 14:54:22 +08:00
liaoxingyu 5ea89bf607 fix bug in sampler
Summary: remove set, which will lead to a bug in random choice.
2020-08-21 14:54:17 +08:00
Xingyu Liao 8f65597d18
update pytorch version 2020-08-21 11:20:00 +08:00
liaoxingyu 8d4bd77ae3 fix visualization problem #233 2020-08-20 17:03:43 +08:00
liaoxingyu 07872d4cdb fix caffe export bug
Summary: put tensor to gpu device
2020-08-20 16:28:52 +08:00
liaoxingyu f4305b0964 fix bnneck 2020-08-20 16:21:14 +08:00
liaoxingyu ac8409a7da updating for pytorch1.6 2020-08-20 15:51:41 +08:00
liaoxingyu 9844760d7f fix naive sampler time-consuming problem
Summary: list.remove has O(n) time complexity and use set instead.
2020-08-20 15:44:26 +08:00
liaoxingyu 2c0320da1a fix bug in pid_dict
Summary: id index will be different in different machine.
This will lead to different fc layer when training on different machine.
2020-08-20 15:41:14 +08:00
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