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

200 Commits (0da5917064c1f09f33753a7aeae2b71a9adb4410)

Author SHA1 Message Date
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
Xingyu Liao 15c556c43a
remove apex dependency (#442) 
Summary: Use Pytorch1.6(or above) built-in amp training
 2021-03-23 12:12:35 +08:00
Xingyu Liao 883fd4aede
add configurable decorator & linear loss decouple (#441) 
Summary: Add configurable decorator which can call `Baseline` with `Baseline(cfg)` or `Baseline(cfg, heads=heads, ...)`
Decouple linear and loss computation for partial-fc support.

Reviewed By: l1aoxingyu
 2021-03-23 12:10:06 +08:00
Xingyu Liao 9b5af4166e
support repvgg (#429) 
Summary:
* support repvgg backbone, and verify the consistency of train mode and eval mode
* onnx export logger style modification
 2021-03-11 14:23:24 +08:00
liaoxingyu f57c5764e3 support multi-node training 2021-03-09 20:07:28 +08:00
liaoxingyu 68c190b53c replace list in evaluator process with dict 2021-03-09 20:07:13 +08:00
liaoxingyu 96fd58c48f update vehicleid and veriwild datasets 2021-02-26 19:26:48 +08:00
liaoxingyu 52b75b7974 update prid and grid datasets 2021-02-26 19:26:38 +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 6b4b935ce4 fix augmix warning 
Summary: add array.clone to avoid warning about numpy array not writable
 2021-01-29 11:50:33 +08:00
liaoxingyu b786001ebd add mobilenetv2 support 2021-01-26 18:17:01 +08:00
liaoxingyu ef6ebf451b refactor apex import 2021-01-23 15:35:48 +08:00
liaoxingyu a53fd17874 update docs 2021-01-23 15:25:58 +08:00
liaoxingyu b5c3c0a24d update docs 2021-01-22 21:11:19 +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 7e83d3175f update README.md 
Summary: add information about fastreid V1.0
 2021-01-18 11:44:55 +08:00
liaoxingyu 15e1729a27 update fastreid V1.0 2021-01-18 11:36:38 +08:00
liaoxingyu c49414bb9f fix SE layer of basicblock in resnet (#375) 
Summary: SE layer defined in `__init__` but not used in `forward`

close #375
 2021-01-04 10:46:54 +08:00
liaoxingyu 2c17847980 feat: freeze FC 
Summary: update freeze FC in the last stages of training
 2020-12-28 14:46:28 +08:00
liaoxingyu fe2e46d40e fix arcSoftmax fp16 training problem 
Summary: fixup fp16 training when using arcSoftmax by aligning the data type
 2020-12-28 14:45:26 +08:00
liaoxingyu 20a01f2545 fix grayscale image input problem 
Summary: handle grayscale images mixed in RGB images by adding the image channels
 2020-12-28 14:45:09 +08:00
liaoxingyu 8083547613 add kwargs for convenient dataset parameters passing (#290) 
Summary: make it more easy for passing dataset kwargs through `build_reid_train_loader`
 2020-12-28 14:39:08 +08:00
liaoxingyu 0e1b91f74a fix checkpoint bug in ddp training 
Summary: change pytorch ddp to apex ddp in checkpoint
 2020-12-28 14:35:02 +08:00
liaoxingyu f56ca8345e fix keywords error 
Summary: add `freeze_fc` and `flip_test` keywords
 2020-12-28 14:34:18 +08:00
Xingyu Liao 0428a270e3
update veri dataset 
Summary: update veri dataset reference information

Reviewed By: l1aoxingyu
 2020-12-22 18:01:37 +08:00
liaoxingyu 766f309d03 feat: update pairwise cosface and pairwise circle loss 2020-12-22 15:51:49 +08:00
liaoxingyu 66941cf27a feat: support flip testing 2020-12-22 15:50:50 +08:00
liaoxingyu bb7a00e615 feat: add save best model checkpoint 2020-12-22 15:50:23 +08:00
liaoxingyu 5469e8ce76 feat: add save best model mechanism 2020-12-22 15:49:46 +08:00
liaoxingyu 04fe9fb2d8 add saivt dataset 2020-12-22 15:47:08 +08:00
liaoxingyu e8739a667e update AirportALERT dataset 
Summary: add more IDs from the single camera
 2020-12-22 15:36:50 +08:00
Xinchen Liu 218a007b6a
Update veri.py 2020-12-21 16:19:06 +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 2724515fd9 save class number to config (#281) 
Summary: Save the class number calculated based on datasets to the config file. If you hard-code the class number, make it unchanged.
 2020-11-06 16:07:37 +08:00
liaoxingyu 7e9a4775da fixup finetune problem 
Summary: support finetune from the other model with different number of classes, and simplify calling way (#325)

close #325

close #325
 2020-11-06 15:58:22 +08:00
liaoxingyu f496193f17 change cross_entroy_loss input name 
Summary: change `pred_class_logits` to `pred_class_outputs` to prevent misleading. (#318)

close #318
 2020-11-06 14:16:31 +08:00
liaoxingyu2 a00e50d37f fix triplet ddp training 
Summary: fixup precision alignment when triplet loss with ddp
 2020-11-06 11:01:10 +08:00
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 bd395917a8 fix splat layer problem (#297) 
Summary: fix get_norm problem in splat.py
 2020-10-09 11:21:36 +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 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