Commit Graph

291 Commits (e0cf8ac56e4b2e99e7cc402901db691a358923d2)
 

Author SHA1 Message Date
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
liaoxingyu 57ebf75f83 fix bug about changing cfg values 2020-08-10 10:29:24 +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
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 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
liaoxingyu 3b57dea49f support regnet backbone 2020-07-17 19:13:45 +08:00
liaoxingyu eb88076714 update tabulate vis 2020-07-15 17:12:24 +08:00
liaoxingyu cf06c9e8c7 update tabulate vis with dataset name 2020-07-15 15:53:03 +08:00
liaoxingyu e9b0198502 update issue template 2020-07-15 15:10:31 +08:00
liaoxingyu 1cf580b6b0 update partial code 2020-07-15 15:08:53 +08:00
liaoxingyu e0fc15269c update tabulate vis 2020-07-15 14:56:18 +08:00
liaoxingyu ef92e63f34 update issue template 2020-07-15 14:56:08 +08:00
liaoxingyu 3f35eb449d minor update 2020-07-14 11:58:06 +08:00
Lingxiao He 7fbdf1fe82
Update dsr_evaluation.py 2020-07-14 11:35:03 +08:00
Lingxiao He c03dccacc5
Update README.md 2020-07-14 11:33:27 +08:00
Lingxiao He eb0d5a1dc7
Create CrowdReID 2020-07-13 16:22:09 +08:00
Lingxiao He 6d2ec2cd9e
Update partial_market.yml 2020-07-12 14:06:44 +08:00
Lingxiao He 64de63a9ff
Update partial_market.yml 2020-07-12 13:52:39 +08:00
Lingxiao He ddc9dbf501
Update partial_market.yml 2020-07-12 13:44:22 +08:00
Lingxiao He 115de22532
Update dsr_distance.py 2020-07-12 13:31:21 +08:00
liaoxingyu f8d468647c resnet expansion add 2020-07-10 22:40:07 +08:00
liaoxingyu e81b13798c change way of loss function
Summary: move loss computation from meta_arch to run_step considering distillation loss
2020-07-10 16:28:53 +08:00
liaoxingyu dcd6f72235 partialreid bugfix #161 2020-07-10 16:27:22 +08:00
liaoxingyu ea8a3cc534 fix typro 2020-07-10 16:26:35 +08:00
liaoxingyu ab4afdb3c1 Merge branch 'master' of github.com:L1aoXingyu/fast-reid 2020-07-09 12:10:54 +08:00
liaoxingyu 59a74de240 update vehicle reid config 2020-07-09 12:10:38 +08:00
Xingyu Liao b01bc12a00
Fix the typing error in VeRi
reviewed by: l1aoxingyu
2020-07-07 11:30:15 +08:00
grimoire 76afdd8d66 fix bugs of rank_cylib 2020-07-07 11:12:46 +08:00
liaoxingyu 6cbca133a8 update readme 2020-07-06 17:16:44 +08:00
liaoxingyu 13bb8d5b1c update readme 2020-07-06 17:04:21 +08:00
liaoxingyu fec7abc461 finish v0.2 ddp training 2020-07-06 16:57:43 +08:00
liaoxingyu 5ae2cff47e fix circle/arcface pred_logits
fix #136
2020-07-06 16:57:03 +08:00