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

155 Commits (3880b23ab432078e3fb548ecca5af96b25a58961)

Author SHA1 Message Date
zuchen.wang 3880b23ab4 refactor evaluator 2021-10-27 16:03:43 +08:00
zuchen.wang d4e2ac32d8 add pcb 2021-10-27 15:09:57 +08:00
zuchen.wang 31d7a51b62 fix bug in online_dataset.py 2021-10-22 13:31:12 +08:00
zuchen.wang ae842114c7 update dataset config in trainer.py 2021-10-21 10:50:38 +08:00
zuchen.wang 0dd493b7b4 make pair dataset negative sample from full set 2021-10-20 17:04:27 +08:00
zuchen.wang 6ab4ac9784 modify logger 2021-10-18 17:38:08 +08:00
zuchen.wang e4c46cbfed add r101-se-ibn.yaml 2021-10-18 17:37:41 +08:00
zuchen.wang 3788c5484b add excel data set 2021-10-18 15:37:25 +08:00
zuchen.wang 3e753612a9 change pair evaluator 2021-10-18 15:36:54 +08:00
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