432 Commits

Author SHA1 Message Date
liaoxingyu2
64bf78afee update rank_cylib 2020-11-06 11:00:13 +08:00
liaoxingyu2
3bd2fad9a5 support faiss-based rerank
Summary: accelerate rerank with faiss-gpu
2020-11-06 10:59:53 +08:00
liaoxingyu2
42cadaeebc update backbone and config
Summary: update resnet backbone for adaptation caffe export; modify effnet loading keyword
2020-11-06 10:58:38 +08:00
liaoxingyu
a5dac37bea update bohb search space 2020-10-12 20:21:47 +08:00
liaoxingyu
e5c3ff3b5b fix typro in hpo training 2020-10-10 18:55:32 +08:00
liaoxingyu
8d55a0bbd4 add more hpo search algorithms
Summary: support hyperopt, bohb and population based training
2020-10-10 17:35:25 +08:00
liaoxingyu
bd395917a8 fix splat layer problem (#297)
Summary: fix get_norm problem in splat.py
2020-10-09 11:21:36 +08:00
liaoxingyu
caf11b5cc6 add checkpoint in hpo 2020-10-01 18:12:51 +08:00
liaoxingyu
008e4f700a Merge remote-tracking branch 'origin/master' 2020-10-01 18:10:19 +08:00
liaoxingyu
85cebe311f support hpo project
Summary: support BOHB search algorithm for reid.
2020-10-01 18:10:02 +08:00
liaoxingyu
4629172a90 update distill reid readme 2020-10-01 18:09:18 +08:00
Xingyu Liao
ad9046c46e
add conda install (#259)
support quick set up with conda
2020-10-01 11:23:32 +08:00
liaoxingyu
bd5aaf386a add student model load path 2020-09-30 16:45:46 +08:00
liaoxingyu
69f4407267 update distill reid project (#287)
Summary: refactor distill reid code and config files
2020-09-30 16:05:58 +08:00
liaoxingyu
3d1bae9f13 fix triplet loss backward propagation on multi-gpu training (#82)
Summary: fix torch.distributed.all_gather has no gradient when performing all_gather operation on the provided tensors, instead using `GatherLayer`.
2020-09-28 17:16:51 +08:00
liaoxingyu
10cbaab155 support finetuning from trained models
Summary: add a flag for supporting finetuning model from the trained weights, and it's very useful when performing across domain reid
2020-09-28 17:10:10 +08:00
liaoxingyu
3024cea3a3 add more datasets
Summary: add wildtracker.py datasets and cuhk_sysu.py datasets.
2020-09-28 17:05:40 +08:00
liaoxingyu
bd182691b9 fix readme 2020-09-27 10:52:43 +08:00
liaoxingyu
fae128a4db fix bug in evaluator (#284)
Summary: Change `Trainer` to `DefaultTrainer` in `tools/train.py` for adapting to new version of `build_evaluator`.
2020-09-25 15:05:56 +08:00
liaoxingyu
a25d8a6bc1 refactor code for partial reid (#277)
Summary: make partial reid adapted for new code style

close #277
2020-09-25 14:24:48 +08:00
liaoxingyu
f2d2467ead update projects homepage 2020-09-25 11:33:16 +08:00
liaoxingyu
ea06ead9e3 fix problem when converting to caffe model (#261)
Summary: caffe cannot support normalization(sub mean and div std) in network, so make the normalization operators inplace which will not trigger when converting to caffe.

close #261
2020-09-25 11:26:20 +08:00
liaoxingyu
5dfe537515 update attribute project 2020-09-23 19:45:13 +08:00
liaoxingyu
5b88736e1d refactor evaluation code 2020-09-23 19:32:40 +08:00
liaoxingyu
154a06b875 refactor code 2020-09-23 19:31:46 +08:00
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
Lingxiao He
cf246e9bee
Delete CrowdReID
delete
2020-09-11 15:53:17 +08:00
liaoxingyu
c9824be7e1 fix non-local inter_channels 2020-09-10 11:40:13 +08:00
liaoxingyu
1f4e4909eb fix typro ##255
fix #255
2020-09-10 11:10:40 +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
3c48bf78c1 change config file 2020-09-10 10:58:56 +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
wangguanan
7e2bc6873f support set up with conda 2020-09-08 11:14:05 +08:00
liaoxingyu
57710a9df7 add docs for amp training 2020-09-02 18:07:54 +08:00
liaoxingyu
53fed7451d feat: support amp training
Summary: support automatic mixed precision training #217
2020-09-02 18:03:12 +08:00
liaoxingyu
78f2166596 refactor mgn config 2020-09-02 18:01:07 +08:00
liaoxingyu
0d2c477060 update mgn config 2020-09-01 16:16:40 +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
c1dc4919ef update model zoo 2020-08-31 18:23:14 +08:00
liaoxingyu
1d6e970260 fix typro #229
fix #229
2020-08-26 23:26:01 +08:00