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

133 Commits (d00ce8fc3c71d6043091d6c1e19a3e685c433e5b)

Author SHA1 Message Date
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
liaoxingyu 5ae2cff47e fix circle/arcface pred_logits 
fix #136
 2020-07-06 16:57:03 +08:00
liaoxingyu b7a2b1c21a support DDP sampler and augment 2020-07-06 16:55:23 +08:00
liaoxingyu c851c7edc8 fix dataset prefix #76 2020-07-06 16:54:54 +08:00
liaoxingyu 4cc157a12f add lars optimizer 
Summary: support lars optimizer refer to https://arxiv.org/abs/1708.03888 which can make large batch training stable
 2020-06-22 12:12:19 +08:00
liaoxingyu f10ce253f1 refactor arcface and circle loss 
#111
 2020-06-22 11:56:39 +08:00
liaoxingyu 3840f3f79a fix arcface NaN problem 
Summary: fix classifier init bugs, which will not initialize classifier weights when use arcface or circle loss.
In this way, it will lead loss NaN problem.
 2020-06-18 12:05:44 +08:00
liaoxingyu 8879db3fba update training instruction 
Summary: update dataset configuration and training instruction
 2020-06-16 19:43:36 +08:00
liaoxingyu ecc2b1a790 update naive sampler 
Summary: update naive sampler which will introduce unbalanced sampling
 2020-06-15 20:50:25 +08:00
liaoxingyu 56a1ab4a5d update fast global avgpool 
Summary: update fast pool according to https://arxiv.org/pdf/2003.13630.pdf
 2020-06-12 16:34:03 +08:00
liaoxingyu cbdc01a1c3 update pairwise circle loss 
Summary: add param of pairwise circle loss to config, and update pairwise circle loss version
 2020-06-10 19:07:29 +08:00
liaoxingyu 96b9ad2d99 fix docs in ToTensor 2020-06-10 19:06:03 +08:00
liaoxingyu 3732f94405 update osnet 2020-06-09 14:38:49 +08:00
liaoxingyu 25a7f82df7 change style in baseline 2020-06-05 11:23:11 +08:00
liaoxingyu bc221cb05f fix mgn multi-gpu training problem 
Summary: norm_type in pool_reduce will not change when use syncBN
 2020-06-05 11:11:50 +08:00