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

146 Commits (71903485689de748cf74264473c360afb6accecb)

Author SHA1 Message Date
zuchen.wang 7190348568 fix pair_dataset random seed in val and test mode 
add ce loss
 2021-10-16 21:42:32 +08:00
zuchen.wang 1feda07ce6 evaluation success 2021-10-16 21:24:16 +08:00
zuchen.wang 786c442391 add bughook in train_net.py 2021-10-16 20:56:01 +08:00
zuchen.wang d87f4a8419 add evaluator in fastshoe trainner 2021-10-16 00:03:00 +08:00
zuchen.wang d162c5fc4e add PairHead 2021-10-15 21:26:16 +08:00
zuchen.wang 8c09ec0d3d update forward 2021-10-13 20:05:08 +08:00
zuchen.wang d2ee44d503 change cls_target to clas_target 2021-10-13 20:05:08 +08:00
zuchen.wang 1bd41a79fc add cls target in pair dataset 2021-10-13 20:05:07 +08:00
zuchen.wang 8b309b0f4e add contrastive loss 2021-10-13 20:05:07 +08:00
zuchen.wang 7201a82840 add pair collator_fn 2021-10-13 20:05:07 +08:00
zuchen.wang bced0d04ff override some method in dataset 2021-10-13 20:05:07 +08:00
zuchen.wang e9cd5311f5 add some paths 2021-10-13 20:05:07 +08:00
zuchen.wang 46c1ce83a7 add fastshoe 2021-10-13 20:05:07 +08:00
zuchen.wang 045b1a88bc fix import error in fastclas 
add bughook.py
 2021-10-13 18:07:44 +08:00
l1aoxingyu f4551a128b feat(projects): Add DG-ReID project 2021-09-08 23:10:47 +08:00
xiaomingzhid ee3818e706
Add Semi-Supervised Domain Generalizable Person Re-Identification 2021-09-08 21:11:32 +08:00
Sherlock ced654431b fix bug in attribute evaluation 
# 535
 2021-08-12 14:17:44 +08:00
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