mirrors
/
fast-reid
mirror of https://github.com/JDAI-CV/fast-reid.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
354 Commits
1 Branch
4 Tags
15 MiB
Commit Graph

147 Commits (2f29228086822cf56bdd5cdb368b56f3ed0c7dfc)

Author SHA1 Message Date
liaoxingyu 2f29228086 fix regnet cfg problem 2020-09-23 14:41:44 +08:00
liaoxingyu df823da09a fix typro bce loss 
close #273
 2020-09-18 10:39:10 +08:00
liaoxingyu d9a63a959f fix bug in mgn #272 
Summary: fix get_norm bug in mgn
 2020-09-17 18:17:53 +08:00
liaoxingyu 4fa3f08a4a fix typro 
close #268
 2020-09-14 11:34:28 +08:00
liaoxingyu c9824be7e1 fix non-local inter_channels 2020-09-10 11:40:13 +08:00
liaoxingyu 74a6938289 fix typro 2020-09-10 11:04:59 +08:00
liaoxingyu 648198e6e5 add efficientnet support 2020-09-10 11:04:52 +08:00
liaoxingyu 296fbea989 extent gem pool 2020-09-10 11:01:22 +08:00
liaoxingyu 1b84348619 remove `num_splits` in batchnorm 
Summary: `num_splits` works for GhostBN, but it's very uncommon
 2020-09-10 11:01:07 +08:00
liaoxingyu 4d573b8107 refactor reid head 
Summary: merge BNneckHead, LinearHead and ReductionHead into EmbeddingHead
because they are highly similar and can be prepared for ClsHead
 2020-09-10 10:57:37 +08:00
liaoxingyu 77caa01e34 fix memory leak in sgd #217 2020-09-10 10:50:22 +08:00
liaoxingyu aa5c422606 fix pair-wise circle loss 
fix #252
 2020-09-09 15:28:52 +08:00
liaoxingyu 919a515eb7 fix combine all training 2020-09-09 15:28:21 +08:00
liaoxingyu 53fed7451d feat: support amp training 
Summary: support automatic mixed precision training #217
 2020-09-02 18:03:12 +08:00
liaoxingyu d00ce8fc3c refactor model arch 2020-09-01 16:14:45 +08:00
liaoxingyu 866a196d19 add more datasets support 2020-09-01 16:13:12 +08:00
liaoxingyu f06c8b2ed9 fix bug: make CJ works #239 
fix #239
 2020-08-25 11:31:49 +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
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 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 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 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 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 65169b40bd support ddp testing 2020-07-30 20:15:28 +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 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 e0fc15269c update tabulate vis 2020-07-15 14:56:18 +08:00
liaoxingyu 3f35eb449d minor update 2020-07-14 11:58:06 +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 ea8a3cc534 fix typro 2020-07-10 16:26:35 +08:00
grimoire 76afdd8d66 fix bugs of rank_cylib 2020-07-07 11:12:46 +08:00
liaoxingyu fec7abc461 finish v0.2 ddp training 2020-07-06 16:57:43 +08:00