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
