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

532 Commits (master)
 

Author SHA1 Message Date
darrenhsieh ebc375e51e fix: cmake bug 2021-01-31 18:08:25 +08:00
liaoxingyu 1666c82db4 update readme about FastRT 2021-01-30 16:11:08 +08:00
Xingyu Liao d2f9450041
Merge pull request #397 from TCHeish/FastRTv001 
feat: support reid model with trt C++APIs
 2021-01-30 16:05:05 +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 6b4b935ce4 fix augmix warning 
Summary: add array.clone to avoid warning about numpy array not writable
 2021-01-29 11:50:33 +08:00
Xingyu Liao 1c08b7ac88
update veri-wild training config 
Update bagtricks_R50-ibn.yml in veri-wild
 2021-01-28 11:30:44 +08:00
Xinchen Liu 885dc96608
Update bagtricks_R50-ibn.yml 
The config that obtains the best performance on VeRi-Wild. The model is trained on 4 P40 GPUs
 2021-01-28 11:15:57 +08:00
liaoxingyu 1ed1a13eed update vehicle reid pretrained model (#321, #351) 
close #321, #351
 2021-01-27 17:23:41 +08:00
liaoxingyu 69454860d1 Merge branch 'master' of github.com:L1aoXingyu/fast-reid 2021-01-26 18:17:25 +08:00
liaoxingyu b786001ebd add mobilenetv2 support 2021-01-26 18:17:01 +08:00
Xingyu Liao 63ed07ab0d
update readme 2021-01-25 13:43:04 +08:00
Xinchen Liu efcc2f28cb
Update README.md 
Correct the link to INSTALL.md
 2021-01-25 11:06:19 +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 274cd81dab update docs 2021-01-22 21:06:30 +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 fdaa4b1a84 update readme 2021-01-18 13:51:02 +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
Xingyu Liao 15213dde4b
add workflow to lint Python code 
Summary: merge PR about linting Python code
 2021-01-16 11:25:49 +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 63c952f3cf remove extra file 2020-12-31 10:53:24 +08:00
liaoxingyu ce86c6959d upload fastreid logo 2020-12-30 16:15:06 +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 130199fdce remove invalid config 2020-12-07 14:23:43 +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 1b9799f601 fix demo eval bug (#345) 
Summary: Fix bug in demo.py when computing cmc and mAP in `evaluate_rank`. This is because after changing eval function, the passing params in demo are not changed.

close #345
 2020-12-04 11:14:18 +08:00
Christian Clauss 96aadfa707
Update lint_python.yml 2020-11-26 14:28:52 +01: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