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

596 Commits (3cad97418b27d8c66f9748f6b8ff6a44b69d6274)
 

Author SHA1 Message Date
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
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 a5dac37bea update bohb search space 2020-10-12 20:21:47 +08:00
liaoxingyu e5c3ff3b5b fix typro in hpo training 2020-10-10 18:55:32 +08:00
liaoxingyu 8d55a0bbd4 add more hpo search algorithms 
Summary: support hyperopt, bohb and population based training
 2020-10-10 17:35:25 +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 caf11b5cc6 add checkpoint in hpo 2020-10-01 18:12:51 +08:00
liaoxingyu 008e4f700a Merge remote-tracking branch 'origin/master' 2020-10-01 18:10:19 +08:00
liaoxingyu 85cebe311f support hpo project 
Summary: support BOHB search algorithm for reid.
 2020-10-01 18:10:02 +08:00
liaoxingyu 4629172a90 update distill reid readme 2020-10-01 18:09:18 +08:00
Xingyu Liao ad9046c46e
add conda install (#259) 
support quick set up with conda
 2020-10-01 11:23:32 +08:00
liaoxingyu bd5aaf386a add student model load path 2020-09-30 16:45:46 +08:00
liaoxingyu 69f4407267 update distill reid project (#287) 
Summary: refactor distill reid code and config files
 2020-09-30 16:05:58 +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 bd182691b9 fix readme 2020-09-27 10:52:43 +08:00
liaoxingyu fae128a4db fix bug in evaluator (#284) 
Summary: Change `Trainer` to `DefaultTrainer` in `tools/train.py` for adapting to new version of `build_evaluator`.
 2020-09-25 15:05:56 +08:00
liaoxingyu a25d8a6bc1 refactor code for partial reid (#277) 
Summary: make partial reid adapted for new code style

close #277
 2020-09-25 14:24:48 +08:00
liaoxingyu f2d2467ead update projects homepage 2020-09-25 11:33:16 +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
liaoxingyu 5dfe537515 update attribute project 2020-09-23 19:45:13 +08:00
liaoxingyu 5b88736e1d refactor evaluation code 2020-09-23 19:32:40 +08:00
liaoxingyu 154a06b875 refactor code 2020-09-23 19:31:46 +08:00
liaoxingyu 2f29228086 fix regnet cfg problem 2020-09-23 14:41:44 +08:00
liaoxingyu df823da09a fix typro bce loss 
close #273
 2020-09-18 10:39:10 +08:00
liaoxingyu d9a63a959f fix bug in mgn #272 
Summary: fix get_norm bug in mgn
 2020-09-17 18:17:53 +08:00
liaoxingyu 4fa3f08a4a fix typro 
close #268
 2020-09-14 11:34:28 +08:00
Lingxiao He cf246e9bee
Delete CrowdReID 
delete
 2020-09-11 15:53:17 +08:00
liaoxingyu c9824be7e1 fix non-local inter_channels 2020-09-10 11:40:13 +08:00
liaoxingyu 1f4e4909eb fix typro ##255 
fix #255
 2020-09-10 11:10:40 +08:00
liaoxingyu 74a6938289 fix typro 2020-09-10 11:04:59 +08:00
liaoxingyu 648198e6e5 add efficientnet support 2020-09-10 11:04:52 +08:00
liaoxingyu 296fbea989 extent gem pool 2020-09-10 11:01:22 +08:00
liaoxingyu 1b84348619 remove `num_splits` in batchnorm 
Summary: `num_splits` works for GhostBN, but it's very uncommon
 2020-09-10 11:01:07 +08:00