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

129 Commits (7e652fea2ac76db27170ce8ab294724ddfe8b716)

Author SHA1 Message Date
liaoxingyu 7e652fea2a feat: Add contiguous parameters support 
Support contiguous parameters to train faster. It can split parameters into different contiguous groups by freeze_layer, lr and weight decay.
 2021-07-05 11:10:37 +08:00
liaoxingyu 7c1269d6c3 minor fix for configs 
#519
 2021-06-24 14:26:51 +08:00
liaoxingyu 2fee8327fa minor fix for configs 
#519
 2021-06-23 17:49:35 +08:00
liaoxingyu c3ac4f504c Support amp and resume training in fastface 
AMP in partial-fc needs to be done only on backbone; In order to impl `resume training`, need to save & load different part of classifier weight in each GPU.
 2021-05-31 17:30:43 +08:00
liaoxingyu 91ff631184 Minor changes 
Some minor changes, such as class name changing, remove extra blank line, etc.
 2021-05-31 17:27:14 +08:00
liaoxingyu 8ab3554958 Support self-distill with EMA updated model 2021-05-31 17:17:24 +08:00
Darren 37ccd3683d
unify gen_wts.py and inference.cpp dummy test value. (#457) 
Reviewed by: @L1aoXingyu
 2021-04-19 10:23:15 +08:00
Darren 8276ccf4fd
[v005] set INT8 calibrate set via cmake (#459) 
Reviewed by: @L1aoXingyu
 2021-04-19 10:22:48 +08:00
liaoxingyu e124a9afd3 fix lint_python 2021-04-12 15:08:39 +08:00
Darren e0ad8c70bc
update dockerfile in fastrt (#437) 
Reviewed by: @L1aoXingyu
 2021-04-12 15:06:13 +08:00
慕湮 fc67350e99
Add python interface by pybind11 and Int8 mode 
Reviewed by: @TCHeish
 2021-04-12 15:05:21 +08:00
liaoxingyu 0da5917064 bugfix for ClasDataset 2021-04-08 11:01:15 +08:00
liaoxingyu 44cee30dfc update fastreid v1.2 
Summary:
1. refactor dataloader and heads
2. bugfix in fastattr, fastclas, fastface and partialreid
3. partial-fc supported in fastface
 2021-04-02 21:33:13 +08:00
liaoxingyu 9288db6303 add filp in fasttune config 2021-03-31 17:09:34 +08:00
Xingyu Liao fb36b23678
bugfix for attribute project (#450) 
Summary: refactor sample weight in attribute recognition;
change all options to False in defaults.py and modify yaml files
 2021-03-31 17:07:19 +08:00
Xingyu Liao be0a089e1f
bugfix & merge classification transforms (#448) 
Summary: change heads definition in project and config file, merge classification transforms into default transforms
 2021-03-30 15:47:14 +08:00
Xingyu Liao 664ba4ae11
save idx2class dict when training classification (#444) 2021-03-26 20:51:52 +08:00
Xingyu Liao 890224f25c
support classification in fastreid (#443) 
Summary: support classification and refactor build_dataloader which can support explicit parameters passing
 2021-03-26 20:17:39 +08:00
Darren d7c1294d9e
fastrt patch update 
Summary:
move div255 to gpu
add read/write numpy ndarray, which will make the comparison between torch and trt results more easily.

Reviewed By: l1aoxingyu
 2021-03-10 10:53:18 +08:00
darrenhsieh b9bda486f0 add trt api div255 2021-02-27 16:45:29 +08:00
darrenhsieh 69eb044b81 [v004] de-couple fastreid config from Model base-class 
fix baseline class ctr bug
 2021-02-27 16:40:04 +08:00
Xingyu Liao cf46e5f071
Attribute projects update 
Summary: 1.fix zero divided in attribute metric computation;2.update market/duke attribute dataset loading.
Reviewed by: l1aoxingyu
 2021-02-18 10:32:48 +08:00
darrenhsieh 0617e3eeb7 refactor: embedding_head 
* add abstract poolinglayer

* add poolinglayer factory
 2021-02-13 21:28:29 +08:00
darrenhsieh 159494e4a4 refactor: model class 
* add Module as base class

* add head type

* replace template func args with Module ptr

* add Module factory
 2021-02-13 21:14:14 +08:00
Jinkai Zheng 39e25e0457 add Market1501Attr and DukeMTMCAttr dataloader 2021-02-05 22:36:54 +08:00
Jinkai Zheng c9537c97d1 add eps in attr_evaluation.py 2021-02-05 21:52:39 +08:00
darrenhsieh 5f7d3d586e refactor: model.h 2021-01-31 18:12:35 +08:00
darrenhsieh ebc375e51e fix: cmake bug 2021-01-31 18:08:25 +08:00
darrenhsieh dc5f1924dc feat: support reid model with trt C++APIs 
Summary: support reid model with tensorrt network definition APIs
 2021-01-30 15:40:05 +08:00
liaoxingyu 2f877d239b update fastdistill readme 2021-01-30 11:52:47 +08:00
liaoxingyu 77a91b1204 feat: support multi-teacher kd 
Summary: support multi-teacher kd with logits and overhaul distillation
 2021-01-29 17:25:31 +08:00
liaoxingyu db8670db63 feat: update naic20 1-st solution 2021-01-29 11:50:59 +08:00
liaoxingyu 6ab40bd43a change CrossDomainReID repo name 2021-01-22 11:19:14 +08:00
liaoxingyu e26182e6ec make lr warmup by iter 
Summary: change warmup way by iter not by epoch, which will make it more flexible when training small epochs
 2021-01-22 11:17:21 +08:00
liaoxingyu 15e1729a27 update fastreid V1.0 2021-01-18 11:36:38 +08:00
liaoxingyu a327a70f0d v0.3 update 
Summary:
1. change DPP training in apex way;
2. make warmup scheduler by iter and lr scheduler by epoch;
3. replace random erasing with torchvision implementation;
4. naming modification in config file
 2020-12-07 14:19:20 +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 caf11b5cc6 add checkpoint in hpo 2020-10-01 18:12:51 +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
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 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 5dfe537515 update attribute project 2020-09-23 19:45:13 +08:00
Lingxiao He cf246e9bee
Delete CrowdReID 
delete
 2020-09-11 15:53:17 +08:00
wangguanan dc9d1e39d0 add msmt17 results 2020-08-10 11:04:28 +08:00
wangguanan bbbcdb35ea add msmt17 results 2020-08-10 10:54:42 +08:00