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

244 Commits (6687df06e01893b275be15166edcbe3ffd4ee87c)

Author SHA1 Message Date
zuchen.wang 6687df06e0 update default config 2021-10-27 15:24:25 +08:00
zuchen.wang d4e2ac32d8 add pcb 2021-10-27 15:09:57 +08:00
l1aoxingyu d73fc53f8a feat: let compile evaluation cython automatically 2021-10-22 13:08:33 +08:00
zuchen.wang 5770a8963b fix 2021-10-18 21:12:26 +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 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 d162c5fc4e add PairHead 2021-10-15 21:26:16 +08:00
zuchen.wang 8a746acb01 set find_unused_parameters=True 2021-10-13 20:05:08 +08:00
zuchen.wang ef46486804 train contrastive loss 2021-10-13 20:05:08 +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
xyliao 10a5f38aaa
fix(dataloader): delete background threading when dataloader is freed (#583) 2021-10-01 16:14:36 +08:00
l1aoxingyu d9d6e19b2c fix: fix bugs about import of torch version 2021-09-21 21:42:00 +08:00
ViokingTung 2cac19ce31
fix num_vis type error bug in visualizer.py (#566) 2021-09-02 14:30:18 +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
keith 44d1e04e9a
Added: mobilenetv3 backbone (#525) 2021-06-30 14:49:28 +08:00
liaoxingyu 10b04b75ff Support lazy loading in dataset 
Add an option for lazy loading, which will parse the data when using it. Specifically, when calling `dataset.query`, it will start to parse the query data, and `train` and `gallery` will not be parsed in advanced. This is useful when you want to run on test dataset but train dataset is large.

See examples in market1501.py
 2021-06-17 16:59:02 +08:00
liaoxingyu d792a69f3f bugfix for veriwild and vehicleid 
Change `camids` from string to int when evaluation, otherwise, it will raise AttributeError

#497
 2021-06-17 16:52:47 +08:00
liaoxingyu ac6256887c Merge branch 'master' of github.com:L1aoXingyu/fast-reid 2021-06-17 16:07:17 +08:00
liaoxingyu 62ad5e1a8b bugfix for cuda tensor to numpy 
Move cuda tensor to host memory in predictor which can be used for following process.
 2021-06-17 16:05:45 +08:00
Jinkai Zheng f2286e7f55
Updated the way of reading VERI-Wild's TXT files (#507) 
Reviewed by: @L1aoXingyu
 2021-06-11 13:46:46 +08:00
liaoxingyu 4f90197336 add logger for pretrain model downloading 2021-06-08 15:57:47 +08:00
liaoxingyu 7ed6240e2c fix ReidEvalution too much memory cost 
Move `matches` matrix computation in each iteration to reduce the extra memory cost

#420 #404
 2021-06-08 15:41:43 +08:00
liaoxingyu de81b3dbaa fix ClasEvalution too much memory cost 
Compute total number of correct predictions on each batch avoiding keeping all predicted logits, which will cost too much memory when the number of classes is large

#503
 2021-06-07 15:48:47 +08:00
liaoxingyu 8f8cbf9411 fix lr scheduler warning when amp training 
Skip lr scheduler when this iteration creates NaN gradients
 2021-06-02 16:35:46 +08:00
liaoxingyu 2d2279be6a refactor freeze training condition 2021-06-02 15:55:35 +08:00
liaoxingyu 0572765085 fix for lint_python 2021-05-31 17:40:33 +08:00
liaoxingyu 54f96ba78a fix for lint_python 2021-05-31 17:36:56 +08:00
liaoxingyu 6300bd756e Bugfix for cls_layer 
In `any_softmax`, all operations are in-place, so pass into the `logits.clone()` to prevent outside logits changed.
 2021-05-31 17:32:24 +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
liaoxingyu 256721cfde Impl `freezebb` in optimizer's step() 
Make impl. of `freezebb` consistent with impl. of grad clip, and both are implemented through step() in optimizer
 2021-05-31 17:15:26 +08:00
liaoxingyu 07b8251ccb Support gradient clip 
Follow detectron2's instruction and add gradient clip in step function of optimizer
 2021-05-31 17:11:37 +08:00
liaoxingyu 2cabc3428a Support vision transformer backbone 2021-05-31 17:08:57 +08:00
liaoxingyu 2b65882447 change way of layer freezing 
Remove `find_unused_parameters` in DDP and add a new step function in optimizer for freezing backbone. It will accelerate training speed in this way.
 2021-05-25 15:57:09 +08:00
Sherlock Liao ff8a958fff
bugfix for `plain_train_net.py` and lr scheduler step (#484) 2021-05-11 15:46:17 +08:00
Xingyu Liao 46b0681313
fix randompatch (#470) 2021-04-21 17:08:16 +08:00
liaoxingyu 0c8e3d9805 update imbalanced sampler 
Summary: add a new sampler, which is useful for imbalanced or long-tail dataset. This refers to ufoym/imbalanced-dataset-sampler.
 2021-04-21 17:05:10 +08:00
Xingyu Liao 1dce15efad
faster dataloader with pre-fetch and cuda stream (#456) 
Summary: add a background thread to create a generator with pre-fetch, and create a new cuda stream to copy tensor from cpu to gpu in parallel.

Reviewed by: l1aoxingyu
 2021-04-12 15:03:35 +08:00
liaoxingyu 55300730e1 update fastreid v1.2 readme and changelog 2021-04-06 20:09:13 +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
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
Xie Jingyi 25cfa88fd9
Fix bug: COMBINEALL should add extra prefix to avoid pid conflictions (#447) 2021-03-30 15:49:52 +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 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